--model_file in plot_surface.py should be specified

tomgoldstein / loss-landscape

Code for visualizing the loss landscape of neural nets

MIT License

2.72k stars 388 forks source link

--model_file in plot_surface.py should be specified #11

Open zdhNarsil opened 5 years ago

zdhNarsil commented 5 years ago

Inplot_surface.py, instead of parser.add_argument('--model_file', default='', help='path to the trained model file') I think it should be parser.add_argument('--model_file', default='cifar10/trained_nets/resnet56_sgd_lr=0.1_bs=128_wd=0.0005/model_300.t7', help='path to the trained model file') when the .t7 file of resnet56 trained with cifar10 has been downloaded.

densechen commented 3 years ago

Hi, @zdhNarsil , If we do not specify the model-file, will the program train the model automatically? Or just use the initialize value?