pytorch-code

To train the model

nohup visdom -logging_level 100 >output_visdom.log & (to look to the training progress)

python train_classifier.py

The options are modifiable in options_classifier.py (choose for models, epochs etc.)

python test_classifier.py

Name		Name	Last commit message	Last commit date
Latest commit History 18 Commits
networks		networks
.DS_Store		.DS_Store
README.md		README.md
options_classifier.py		options_classifier.py
sftp-config.json		sftp-config.json
submit_train.py		submit_train.py
train_classifier.py		train_classifier.py
utils.py		utils.py