изменил конфиг обучения

4 weeks ago · bc1f6ce578
parent 3b76f81c70
commit bc1f6ce578
1 changed files with 3 additions and 3 deletions
--- a/train_scripts/config_ensemble.yaml
+++ b/train_scripts/config_ensemble.yaml
@ -1,5 +1,5 @@
 # Training params
-epoch : 10
+epoch : 5
 batch_size : 4
 num_workers: 8
 limit : 5
@ -16,5 +16,5 @@ loss_function :
 # Scheduler
 scheduler :
  name : torch.optim.lr_scheduler.StepLR
-  gamma : 0.01
-  step_size : 10
+  gamma : 0.15
+  step_size : 3