search

rdevon / cortex

A machine learning library for PyTorch
BSD 3-Clause "New" or "Revised" License
92 stars 25 forks source link

Fix device bug #50

Closed Entodi closed 6 years ago

Entodi commented 6 years ago

While reloading models, models were tend to be set as default cuda device, which is wrong to parameters of experiment.

Entodi commented 6 years ago

Not working properly yet

Entodi commented 6 years ago

Now is working