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='
|
| 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='
|
| 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.")
|
|
|
|
| 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.")
|