removed batch dict unpacking

JVGD · Nov 1, 2021 · 6b3c14b · 6b3c14b
1 parent 2eb9816
commit 6b3c14b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/experiment1/model.py b/experiment1/model.py
@@ -35,8 +35,7 @@ def forward(self, x):
 
     def step(self, batch, batch_idx, *args, **kwargs) -> T.Tensor:
         # Unpacking
-        samples = batch["sample"]
-        targets = batch["target"]
+        samples, targets = batch
 
         # Forward
         targets_pred = self(samples)