update

engine.py

@@ -739,7 +739,7 @@ class RTCTrainer(object):
         self.criterion = criterion
         self.world_size = world_size
         self.grad_accumulate = args.grad_accumulate
-        self.clip_grad = 10
+        self.clip_grad = 35
         self.heavy_eval = False
         # weak augmentatino stage
         self.second_stage = False

train_multi_gpus.sh

@@ -7,7 +7,7 @@ python -m torch.distributed.run --nproc_per_node=8 train.py \
                                                     -m rtcdet_m \
                                                     -bs 128 \
                                                     -size 640 \
-                                                    --wp_epoch 5 \
+                                                    --wp_epoch 3 \
                                                     --max_epoch 300 \
                                                     --eval_epoch 10 \
                                                     --no_aug_epoch 20 \