From 3fa8d9b4487f1b9802b3ad950db05b6dfbd9b674 Mon Sep 17 00:00:00 2001
From: Marcin Kostrzewski <komar86311@gmail.com>
Date: Wed, 11 May 2022 19:10:26 +0200
Subject: [PATCH] Floatify

---
 model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model.py b/model.py
index 683b854..6e2aac9 100644
--- 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):