@@ -7,7 +7,7 @@ num_classes: 2
num_keypoints: 2
opt: 'adamw'
batch_size: 2
-epochs: 10
+epochs: 5000
lr: 0.005
momentum: 0.9
weight_decay: 0.0001
