Update Model/MultimodelNER/VLSP2016/train_umt_2016.py
Browse files
Model/MultimodelNER/VLSP2016/train_umt_2016.py
CHANGED
@@ -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='
|
156 |
parser.add_argument('--crop_size', type=int, default=224, help='crop size of image')
|
157 |
-
parser.add_argument('--path_image', default='
|
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.")
|
|
|
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.")
|