ShandaAI
/

FlowSep-hive

sound-separation

Model card Files Files and versions

JusperLee commited on 6 days ago

Commit

7af3360

·

verified ·

1 Parent(s): 8f911c6

Update config.yaml

Files changed (1) hide show

config.yaml +1 -1

config.yaml CHANGED Viewed

@@ -89,7 +89,7 @@ model:
       target: latent_encoder.autoencoder.AutoencoderKL
       params:
         # reload_from_ckpt: "model_logs/pretrained/vae.ckpt"
-        reload_from_ckpt: "models/FlowSep/vae.ckpt"
         batchsize: 2
         monitor: val/rec_loss
         image_key: fbank

       target: latent_encoder.autoencoder.AutoencoderKL
       params:
         # reload_from_ckpt: "model_logs/pretrained/vae.ckpt"
+        reload_from_ckpt: "vae.ckpt"
         batchsize: 2
         monitor: val/rec_loss
         image_key: fbank