awesome-semantic-segmentation-pytorch icon indicating copy to clipboard operation
awesome-semantic-segmentation-pytorch copied to clipboard

Where can I get the checkpoint file?

Open MaverickPeter opened this issue 4 years ago • 2 comments

MaverickPeter avatar Sep 30 '19 19:09 MaverickPeter

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')

vlordier avatar Oct 04 '19 15:10 vlordier

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

pyradd avatar Oct 08 '19 09:10 pyradd