s444463/en-ner-conll-2003
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

unresolved problem with epoch time

Browse Source
This commit is contained in:
Mikolaj Pokrywka 2022-06-06 20:30:23 +02:00
parent 3e13181868
commit 5eab9c5889
1 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
run.py
View File

@ -203,14 +203,6 @@ if __name__ == "__main__":
train_tokens_ids = data_process(X)
train_labels = labels_process(Y)
train_tokens_ids = torch.tensor(train_tokens_ids)
train_tokens_ids = train_tokens_ids.to(device)
# print('train is cuda?', X.is_cuda)
train_labels = torch.tensor(train_labels)
train_labels = train_labels.to(device)
# print('train labels is cuda?', Y.is_cuda)
ner_model = NERModel().to(device)
@ -239,11 +231,11 @@ if __name__ == "__main__":
X = manual_process(X, X_raw_single)
# X = X.to(device)
print('train is cuda?', X.is_cuda)
X = X.to(device)
# print('train is cuda?', X.is_cuda)
# Y = Y.to(device)
print('train labels is cuda?', Y.is_cuda)
Y = Y.to(device)
# print('train labels is cuda?', Y.is_cuda)
Y_predictions = ner_model(X)