Move queue to gpu when resuming checkpoint - SWAV self supervised model

pl_bolts/models/self_supervised/swav/swav_module.py

@@ -183,8 +183,8 @@ def on_train_epoch_start(self):
                     self.feat_dim,
                 )
 
-                if self.gpus > 0:
-                    self.queue = self.queue.cuda()
+            if self.gpus > 0 and self.queue is not None:
+                self.queue = self.queue.cuda()
 
         self.use_the_queue = False