search

gsbDBI / torch-choice

Choice modeling with PyTorch: logit model and nested logit model
MIT License
39 stars 8 forks source link

Add device information to conditional-logit and nested-logit models #8

Closed TianyuDu closed 2 years ago

TianyuDu commented 2 years ago

(1) Add thedevice attribute to conditional-logit and nested-logit models, (2) add the device information to the __str__ method as well; and (3) add the to() method to these models for easier moving them between CPU and GPU.

TianyuDu commented 2 years ago

I am working on this on the device-method branch.

TianyuDu commented 2 years ago

I have initiated a pull request for this feature here.

Merged : ).