AiArtLab
/

simplevae

Model card Files Files and versions

recoilme commited on Sep 15

Commit

41c6022

·

verified ·

1 Parent(s): 5c46d24

Update README.md

Files changed (1) hide show

README.md +4 -1

README.md CHANGED Viewed

@@ -50,4 +50,7 @@ https://imgsli.com/NDE1MzE0/5/2
  - Precision: FP32
  - Effective Batch Size: 16
  - Optimizer: AdamW (8-bit)
- - Balanced losses (lpips, MSE, MAE, Edge, KL)

  - Precision: FP32
  - Effective Batch Size: 16
  - Optimizer: AdamW (8-bit)
+ - Balanced losses (lpips, MSE, MAE, Edge, KL)
+## Source
+https://huggingface.co/AiArtLab/simplevae/blob/main/train_vae.py