search

karkaroff / castor

PyTorch deep learning models by the Data Systems Group at the University of Waterloo
http://castor.ai/
Apache License 2.0
0 stars 0 forks source link

WIP: Incorporate single_label datasets for all models #3

Closed achyudh closed 5 years ago

achyudh commented 5 years ago

Fix evaluate_dataset method for single_label datasets

achyudh commented 5 years ago

Revert change to save_state rather than the entire model

achyudh commented 5 years ago

@Ashutosh-Adhikari yup, with --lr 0.01