devanshkv / fetch

A set of deep learning models for FRB/RFI binary classification.
GNU General Public License v3.0
42 stars 32 forks source link

getting model without internet connection #2

Closed caseyjlaw closed 5 years ago

caseyjlaw commented 5 years ago

The fetch/utils.py defines get_model as a call to download the weights from a server at WVU. It would be nice to be able to set this to a model on disk locally. I am interested in using fetch on a machine that has no internet connection (but I can put files there manually).

devanshkv commented 5 years ago

The model files are listed under this csv and can be downloaded by appending this name to http://psrpop.phys.wvu.edu/download.php?val=. Once downloaded (for now) you can keep the weights under $HOME/.keras/models/. I will add the functionality for the custom path directory in the future.