Upload decoder/decoder_config.json with huggingface_hub
Browse files
decoder/decoder_config.json
CHANGED
|
@@ -62,6 +62,8 @@
|
|
| 62 |
"max_grad_norm": 0.5,
|
| 63 |
"save_every_n_samples": 5000000,
|
| 64 |
"find_unused_parameters": false,
|
|
|
|
|
|
|
| 65 |
"n_sample_images": 10,
|
| 66 |
"device": "cuda:0",
|
| 67 |
"epoch_samples": 10000000,
|
|
|
|
| 62 |
"max_grad_norm": 0.5,
|
| 63 |
"save_every_n_samples": 5000000,
|
| 64 |
"find_unused_parameters": false,
|
| 65 |
+
"warmup_steps": 50,
|
| 66 |
+
"amp": true,
|
| 67 |
"n_sample_images": 10,
|
| 68 |
"device": "cuda:0",
|
| 69 |
"epoch_samples": 10000000,
|