diff --git a/models.py b/models.py
index 293b66e1..3a20fcce 100755
--- a/models.py
+++ b/models.py
@@ -131,6 +131,7 @@ class YOLOLayer(nn.Module):
                 if p.is_cuda:
                     self.grid_xy = self.grid_xy.cuda()
                     self.anchor_vec = self.anchor_vec.cuda()
+                    self.anchor_wh = self.anchor_wh.cuda()
 
         # p.view(bs, 255, 13, 13) -- > (bs, 3, 13, 13, 80)  # (bs, anchors, grid, grid, classes + xywh)
         p = p.view(bs, self.nA, self.nC + 5, nG, nG).permute(0, 1, 3, 4, 2).contiguous()  # prediction
diff --git a/utils/gcp.sh b/utils/gcp.sh
index b0f5c6a9..5700fa26 100755
--- a/utils/gcp.sh
+++ b/utils/gcp.sh
@@ -1,7 +1,7 @@
 #!/usr/bin/env bash
 
 # Start
-sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py --freeze
+sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 && python3 train.py
 
 # Resume
 python3 train.py --resume