From 9ceeae074930340f602a24230224564ba752d687 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 19 Oct 2021 14:57:56 +0200
Subject: [PATCH] Learning rate range test.

---
 climax/main/lr_range_test.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/climax/main/lr_range_test.py b/climax/main/lr_range_test.py
index 99e716e..1bdaa16 100644
--- a/climax/main/lr_range_test.py
+++ b/climax/main/lr_range_test.py
@@ -11,6 +11,7 @@ from datetime import timedelta
 from logging.config import dictConfig
 
 # externals
+import torch
 import xarray as xr
 from sklearn.model_selection import train_test_split
 from torch.utils.data import DataLoader
-- 
GitLab