From ef18b2e6d86221cc948c8a1f73f2a4e408daa96a Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Tue, 16 Apr 2024 18:20:19 +0800 Subject: [PATCH] debug --- configs/vae_3d/inference/16x256x256.py | 4 +--- configs/vae_magvit_v2/inference/16x128x128.py | 2 ++ 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/configs/vae_3d/inference/16x256x256.py b/configs/vae_3d/inference/16x256x256.py index 89591d3..5cda9bd 100644 --- a/configs/vae_3d/inference/16x256x256.py +++ b/configs/vae_3d/inference/16x256x256.py @@ -53,6 +53,4 @@ load = None batch_size = 8 grad_clip = 1.0 -lr = 1e-4 - -calc_loss = False \ No newline at end of file +lr = 1e-4 \ No newline at end of file diff --git a/configs/vae_magvit_v2/inference/16x128x128.py b/configs/vae_magvit_v2/inference/16x128x128.py index 24628ff..b49ac9f 100644 --- a/configs/vae_magvit_v2/inference/16x128x128.py +++ b/configs/vae_magvit_v2/inference/16x128x128.py @@ -80,3 +80,5 @@ load = None batch_size = 4 lr = 1e-4 grad_clip = 1.0 + +calc_loss = False \ No newline at end of file