Testing a trained model throws error

castorini / hedwig

PyTorch deep learning models for document classification

Apache License 2.0

593 stars 125 forks source link

Testing a trained model throws error #32

Open Ashutosh-Adhikari opened 5 years ago

Ashutosh-Adhikari commented 5 years ago

We don't need to search for a snapshot when using the flag --trained-model.

achyudh commented 4 years ago

Across hedwig, the --trained-model arg is used to point towards a snapshot. So to fix this should we add a flag to allow testing directly on pretrained models?