Where can I get the checkpoint file?

[vlordier]

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

[pyradd]

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