checkpoitn save fix for ae norm

This commit is contained in:
2023-07-02 14:59:31 +02:00
parent cd6d2bd1fb
commit ef73cd1a20
2 changed files with 3 additions and 4 deletions
-1
View File
@@ -58,7 +58,6 @@ block_setting_reverse = [
CNBlockConfig(64, 32, 9),
CNBlockConfig(32, 16, 3),
CNBlockConfig(16, 3, 3),
# CNBlockConfig(3, 3, 3),
]
config['cond_ae']['encoder_block_settings'] = block_setting