Minor

models/mask_transformer/transformer.py

@@ -179,7 +179,7 @@ class MaskTransformer(nn.Module):
         clip_model, clip_preprocess = clip.load(clip_version, device='cpu',
                                                 jit=False)  # Must set jit=False for training
         # Cannot run on cpu
-        if str(self.device) != "cpu":
+        if str(self.opt.device) != "cpu":
             clip.model.convert_weights(
                 clip_model)  # Actually this line is unnecessary since clip by default already on float16
             # Date 0707: It's necessary, only unecessary when load directly to gpu. Disable if need to run on cpu
@@ -732,7 +732,7 @@ class ResidualTransformer(nn.Module):
         clip_model, clip_preprocess = clip.load(clip_version, device='cpu',
                                                 jit=False)  # Must set jit=False for training
         # Cannot run on cpu
-        if str(self.device) != "cpu":
+        if str(self.opt.device) != "cpu":
             clip.model.convert_weights(
                 clip_model)  # Actually this line is unnecessary since clip by default already on float16
             # Date 0707: It's necessary, only unecessary when load directly to gpu. Disable if need to run on cpu