type=int,
help='Number of iteration times.')
parser.add_argument('-prew2v',
default=True,
type=bool,
help='If use pre-trained word2vec.')
parser.add_argument('-loss_fn',
default='margin_loss_v1',
type=str,
help='loss functions.')
parser.add_argument('-dataname',
default='aapd',
type=str,
help='Name of dataset.')
parser.add_argument('-bestpath',
default=r'./data/saved/best/best_model.ckpt',
type=str,
help='File Path of the best models.')
cfg = parser.parse_args()
if __name__ == "__main__":
# print(cfg.gpus[0])
model = Model(cfg)
# model.BuildArch()
model.train()
