From f20ca1c0dffcd84a02ff495af9d5a08999e526b3 Mon Sep 17 00:00:00 2001 From: GnSight Date: Sat, 16 Oct 2021 15:32:35 +0800 Subject: [PATCH] fixed file extension --- tutorials/03-advanced/image_captioning/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tutorials/03-advanced/image_captioning/train.py b/tutorials/03-advanced/image_captioning/train.py index 73007637..61cda919 100644 --- a/tutorials/03-advanced/image_captioning/train.py +++ b/tutorials/03-advanced/image_captioning/train.py @@ -72,9 +72,9 @@ def main(args): # Save the model checkpoints if (i+1) % args.save_step == 0: torch.save(decoder.state_dict(), os.path.join( - args.model_path, 'decoder-{}-{}.ckpt'.format(epoch+1, i+1))) + args.model_path, 'decoder-{}-{}.pth'.format(epoch+1, i+1))) torch.save(encoder.state_dict(), os.path.join( - args.model_path, 'encoder-{}-{}.ckpt'.format(epoch+1, i+1))) + args.model_path, 'encoder-{}-{}.pth'.format(epoch+1, i+1))) if __name__ == '__main__':