From acd5250c12f7a837b0dac0381aaf057e251dc846 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ga=C5=A1per=20Spagnolo?= <gasper.spagnolo@outlook.com>
Date: Tue, 1 Aug 2023 14:58:24 +0200
Subject: [PATCH] Move the model to gpu

---
 ft.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ft.py b/ft.py
index 6b85b9e..1a7a8fe 100644
--- a/ft.py
+++ b/ft.py
@@ -21,6 +21,8 @@ class FT:
             self.model_name
         )
 
+        self.model.to(self.device)
+
         # set up optimizer
         self.optimizer = torch.optim.AdamW(self.model.parameters(), lr=1e-5)
         self.scheduler = ReduceLROnPlateau(