jiwan-chung commited on
Commit
901ff30
1 Parent(s): 8a76135
Files changed (1) hide show
  1. arguments.py +1 -1
arguments.py CHANGED
@@ -20,7 +20,7 @@ def get_args():
20
  parser.add_argument(
21
  '--label_path', type=str, default='./data/esper_demo/labels_all.json', help='style label info file path')
22
  parser.add_argument(
23
- '--checkpoint', type=str, default='./data/esper_demo/ckpt/gpt2_style_2', help='checkpoint file path')
24
 
25
  parser.add_argument(
26
  '--prefix_length', type=int, default=10, help='prefix length for the visual mapper')
 
20
  parser.add_argument(
21
  '--label_path', type=str, default='./data/esper_demo/labels_all.json', help='style label info file path')
22
  parser.add_argument(
23
+ '--checkpoint', type=str, default='./data/esper_demo/ckpt/gpt2_style', help='checkpoint file path')
24
 
25
  parser.add_argument(
26
  '--prefix_length', type=int, default=10, help='prefix length for the visual mapper')