Update configs/train_config.json
Browse files
configs/train_config.json
CHANGED
|
@@ -1,10 +1,12 @@
|
|
| 1 |
{
|
| 2 |
"generator_lr" : 1e-4,
|
| 3 |
"discriminator_lr" : 4e-4,
|
|
|
|
| 4 |
"epochs" : 15,
|
| 5 |
"lr_decrease_epoch" : 10,
|
| 6 |
"finetuning_generator_lr" : 1e-6,
|
| 7 |
"finetuning_iterations" : 3500,
|
|
|
|
| 8 |
"batch_size" : 4,
|
| 9 |
"number_of_mults" : 3
|
| 10 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"generator_lr" : 1e-4,
|
| 3 |
"discriminator_lr" : 4e-4,
|
| 4 |
+
"epochs" : 1,
|
| 5 |
"epochs" : 15,
|
| 6 |
"lr_decrease_epoch" : 10,
|
| 7 |
"finetuning_generator_lr" : 1e-6,
|
| 8 |
"finetuning_iterations" : 3500,
|
| 9 |
+
"batch_size" : 1,
|
| 10 |
"batch_size" : 4,
|
| 11 |
"number_of_mults" : 3
|
| 12 |
}
|