Update scheduler.py (#40)

An exception is made where steps are if float
miguelvr · Jul 29, 2020 · 0ecbb63 · 0ecbb63
1 parent 7fb8fbf
commit 0ecbb63
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dropblock/scheduler.py b/dropblock/scheduler.py
@@ -7,7 +7,7 @@ def __init__(self, dropblock, start_value, stop_value, nr_steps):
         super(LinearScheduler, self).__init__()
         self.dropblock = dropblock
         self.i = 0
-        self.drop_values = np.linspace(start=start_value, stop=stop_value, num=nr_steps)
+        self.drop_values = np.linspace(start=start_value, stop=stop_value, num=int(nr_steps))
 
     def forward(self, x):
         return self.dropblock(x)