Update Dockerfile

Dockerfile

@@ -45,8 +45,7 @@ CMD bash -c "accelerate launch \
     --learning_rate=3e-5 \
     --gradient_checkpointing \
     --max_train_steps=7000 \
-    --lr_scheduler=cosine_with_restarts \
-    --lr_scheduler_num_cycles=7 \
+    --lr_scheduler=cosine \
     --lr_warmup_steps=200 \
     --center_crop \
     --random_flip \