diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-22 08:15:58 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-04-22 08:15:58 +0200 |
| commit | 1ca8b0b9e0613c1e02f6a5d8b49e20c4d6916412 (patch) | |
| tree | 5e610ac459c9b254f8826e92372346f01f8e2412 /training/configs/image_transformer.yaml | |
| parent | ffa4be4bf4e3758e01d52a9c1f354a05a90b93de (diff) | |
Fixed training script, able to train vqvae
Diffstat (limited to 'training/configs/image_transformer.yaml')
| -rw-r--r-- | training/configs/image_transformer.yaml | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/training/configs/image_transformer.yaml b/training/configs/image_transformer.yaml index 228e53f..e6637f2 100644 --- a/training/configs/image_transformer.yaml +++ b/training/configs/image_transformer.yaml @@ -85,3 +85,5 @@ trainer: max_epochs: 512 terminate_on_nan: true weights_summary: true + +load_checkpoint: null |