sort=False, shuffle=False) test_iter = data.Iterator(test, batch_size=args.batch_size, device=args.gpu, train=False, repeat=False, sort=False, shuffle=False) config = args config.words_num = len(TEXT.vocab) if args.dataset == 'RelationPrediction': config.rel_label = len(RELATION.vocab) model = RelationPrediction(config) else: print("Error Dataset") exit() model.embed.weight.data.copy_(TEXT.vocab.vectors) if args.cuda: model.cuda() print("Shift model to GPU") print(config) print("VOCAB num", len(TEXT.vocab)) print("Train instance", len(train)) print("Dev instance", len(dev)) print("Test instance", len(test)) print("Relation Type", len(RELATION.vocab))
sort_within_batch=False) test_iter = data.Iterator(test, batch_size=args.batch_size, device=args.gpu, train=False, repeat=False, sort=False, shuffle=False, sort_within_batch=False) config = args config.words_num = len(TEXT.vocab) if args.dataset == 'RelationPrediction': config.rel_label = len(RELATION.vocab) model = RelationPrediction(config) else: print("Error Dataset") exit() model.embed.weight.data.copy_(TEXT.vocab.vectors) if args.cuda: model = model.to(torch.device("cuda:{}".format(args.gpu))) print("Shift model to GPU") print(config) print("VOCAB num", len(TEXT.vocab)) print("Train instance", len(train)) print("Dev instance", len(dev)) print("Test instance", len(test)) print("Relation Type", len(RELATION.vocab))
device=device, train=False, repeat=False, sort=False, shuffle=False, sort_within_batch=False) config = args config.words_num = len(TEXT.vocab) if args.dataset == 'RelationPrediction': config.rel_label = len(RELATION.vocab) if config.relation_prediction_mode.lower() == 'transformer': model = TransformerModel(config) else: model = RelationPrediction(config) else: print("Error Dataset") exit() model.embed.weight.data.copy_(TEXT.vocab.vectors) if args.cuda: model = model.to(torch.device("cuda:{}".format(args.gpu))) print("Shift model to GPU") print(config) print("VOCAB num", len(TEXT.vocab)) print("Train instance", len(train)) print("Dev instance", len(dev)) print("Test instance", len(test)) print("Relation Type", len(RELATION.vocab))