Skip to content
Snippets Groups Projects
Commit 11b58400 authored by piotrmp's avatar piotrmp
Browse files

Bug fix.

parent ab07051d
Branches
1 merge request!1Migration to UD 2.11
......@@ -223,7 +223,7 @@ def pretrain(model, train_dataloader, test_dataloader, epochs, device='cpu'):
optimizer = Adam(model.parameters(), lr=learning_rate)
print("Pretraining")
test_loop_pretraining(test_dataloader, model)
test_loop_pretraining(test_dataloader, model, device)
for t in range(epochs):
print(f"Epoch {t + 1}\n-------------------------------")
train_loop(train_dataloader, model, optimizer, device)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment