Floatify

2022-05-11 19:10:26 +02:00 · 2022-05-11 19:10:26 +02:00 · 3fa8d9b448
commit 3fa8d9b448
parent 5db367b894
1 changed files with 1 additions and 1 deletions
--- a/model.py
+++ b/model.py
@ -24,7 +24,7 @@ class MLP(nn.Module):

    def forward(self, x):
        x = x.view(x.size(0), -1)
-        return self.layers(x)
+        return self.layers(x.float())


 class PlantsDataset(Dataset):