diff --git a/train.py b/train.py index 60fa867d..1dff7feb 100644 --- a/train.py +++ b/train.py @@ -15,7 +15,7 @@ parser.add_argument('-img_size', type=int, default=32 * 13, help='size of each i parser.add_argument('-resume', default=False, help='resume training flag') parser.add_argument('-batch_report', default=False, help='report TP, FP, FN, P and R per batch (slower)') parser.add_argument('-optimizer', default='SGD', help='optimizer') -parser.add_argument('-freeze_darknet53', default=False, help='freeze darknet53.conv.74 layers for first epoch') +parser.add_argument('-freeze_darknet53', default=True, help='freeze darknet53.conv.74 layers for first epoch') parser.add_argument('-var', type=float, default=0, help='optional test variable') opt = parser.parse_args() print(opt) diff --git a/utils/gcp.sh b/utils/gcp.sh index b041bf46..6d0253cb 100644 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -30,3 +30,14 @@ gsutil cp yolov3/weights/latest.pt gs://ultralytics # Copy latest.pt from bucket gsutil cp gs://ultralytics/latest.pt yolov3/weights/latest.pt +# Testing +sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3 +python3 train.py -epochs 1 -var 1 +python3 train.py -epochs 1 -var 2 +python3 train.py -epochs 1 -var 4 +python3 train.py -epochs 1 -var 8 +python3 train.py -epochs 1 -var 16 +python3 train.py -epochs 1 -var 32 +python3 train.py -epochs 1 -var 64 +sudo shutdown +