awesome-semantic-segmentation-pytorch
awesome-semantic-segmentation-pytorch copied to clipboard
Where can I get the checkpoint file?
Take a look here : save location and here location argument
defaults are set here
parser.add_argument('--save-dir', default='~/.torch/models',
help='Directory for saving checkpoint models')
parser.add_argument('--save-epoch', type=int, default=10,
help='save model every checkpoint-epoch')
parser.add_argument('--log-dir', default='../runs/logs/',
help='Directory for saving checkpoint models')
change this : parser.add_argument('--save-dir', default='~/.torch/models', help='Directory for saving checkpoint models')
To: parser.add_argument('--save-dir', default='checkpoints/', help='Directory for saving checkpoint models')
You should see checkpoints directory created in the directory from where you ran the train.py script