From 1ca8b0b9e0613c1e02f6a5d8b49e20c4d6916412 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Thu, 22 Apr 2021 08:15:58 +0200
Subject: Fixed training script, able to train vqvae

---
 training/configs/image_transformer.yaml | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'training/configs/image_transformer.yaml')

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 
-- 
cgit v1.2.3-70-g09d2