Update model test configs

This commit is contained in:
Eren Gölge
2021-06-28 17:03:19 +02:00
parent 98298ee671
commit fcfd95669a
9 changed files with 23 additions and 23 deletions
+2 -2
View File
@@ -88,7 +88,7 @@
// OPTIMIZER
"epochs": 1, // total number of epochs to train.
"clip_grad": 1.0, // Generator gradient clipping threshold. Apply gradient clipping if > 0
"grad_clip": 1.0, // Generator gradient clipping threshold. Apply gradient clipping if > 0
"lr_scheduler": "MultiStepLR", // one of the schedulers from https://pytorch.org/docs/stable/optim.html#how-to-adjust-learning-rate
"lr_scheduler_params": {
"gamma": 0.5,
@@ -107,7 +107,7 @@
// DATA LOADING
"num_loader_workers": 0, // number of training data loader processes. Don't set it too big. 4-8 are good values.
"num_val_loader_workers": 0, // number of evaluation data loader processes.
"num_eval_loader_workers": 0, // number of evaluation data loader processes.
"eval_split_size": 4,
// PATHS