Spaces:

aharley
/

alltracker_demo

Sleeping

aharley commited on Jun 22

Commit

3ec0cab

verified ·

1 Parent(s): 3bb4be2

Upload net34.py

Files changed (1) hide show

nets/net34.py CHANGED Viewed

@@ -194,7 +194,7 @@ class Net(nn.Module):
     def get_fmaps(self, images_, B, T, sw, is_training, nograd_backbone):
         _, _, H_pad, W_pad = images_.shape # revised HW
-        device = images.device
         C, H8, W8 = self.dim*2, H_pad//8, W_pad//8
         if self.no_split:

     def get_fmaps(self, images_, B, T, sw, is_training, nograd_backbone):
         _, _, H_pad, W_pad = images_.shape # revised HW
+        device = images_.device
         C, H8, W8 = self.dim*2, H_pad//8, W_pad//8
         if self.no_split: