Linhz commited on
Commit
141ca4e
·
verified ·
1 Parent(s): 72bea5a

Update Model/MultimodelNER/VLSP2016/train_umt_2016.py

Browse files
Model/MultimodelNER/VLSP2016/train_umt_2016.py CHANGED
@@ -63,7 +63,7 @@ parser.add_argument("--task_name",
63
 
64
  help="The name of the task to train.")
65
  parser.add_argument("--output_dir",
66
- default='/Model/MultimodelNER/VLSP2016/best_model/',
67
  type=str,
68
  help="The output directory where the model predictions and checkpoints will be written.")
69
 
@@ -152,9 +152,9 @@ parser.add_argument('--layer_num1', type=int, default=1, help='number of txt2img
152
  parser.add_argument('--layer_num2', type=int, default=1, help='number of img2txt layer')
153
  parser.add_argument('--layer_num3', type=int, default=1, help='number of txt2txt layer')
154
  parser.add_argument('--fine_tune_cnn', action='store_true', help='fine tune pre-trained CNN if True')
155
- parser.add_argument('--resnet_root', default='/Model/Resnet/', help='path the pre-trained cnn models')
156
  parser.add_argument('--crop_size', type=int, default=224, help='crop size of image')
157
- parser.add_argument('--path_image', default='/Model/MultimodelNER/VLSP2016/Image', help='path to images')
158
  # parser.add_argument('--mm_model', default='TomBert', help='model name') #
159
  parser.add_argument('--server_ip', type=str, default='', help="Can be used for distant debugging.")
160
  parser.add_argument('--server_port', type=str, default='', help="Can be used for distant debugging.")
 
63
 
64
  help="The name of the task to train.")
65
  parser.add_argument("--output_dir",
66
+ default='Model/MultimodelNER/VLSP2016/best_model/',
67
  type=str,
68
  help="The output directory where the model predictions and checkpoints will be written.")
69
 
 
152
  parser.add_argument('--layer_num2', type=int, default=1, help='number of img2txt layer')
153
  parser.add_argument('--layer_num3', type=int, default=1, help='number of txt2txt layer')
154
  parser.add_argument('--fine_tune_cnn', action='store_true', help='fine tune pre-trained CNN if True')
155
+ parser.add_argument('--resnet_root', default='Model/Resnet/', help='path the pre-trained cnn models')
156
  parser.add_argument('--crop_size', type=int, default=224, help='crop size of image')
157
+ parser.add_argument('--path_image', default='Model/MultimodelNER/VLSP2016/Image', help='path to images')
158
  # parser.add_argument('--mm_model', default='TomBert', help='model name') #
159
  parser.add_argument('--server_ip', type=str, default='', help="Can be used for distant debugging.")
160
  parser.add_argument('--server_port', type=str, default='', help="Can be used for distant debugging.")