nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-15 03:15:20 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

debug

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-24 10:12:52 +08:00
parent 988bc3bb65
commit 90775e52bd
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
scripts/train-vae-v2.py
View file

@ -369,7 +369,6 @@ def main():
running_loss += vae_loss.item()
# ====== Discriminator Loss ======
if global_step > cfg.discriminator_start:
# if video_contains_first_frame:
@ -419,6 +418,10 @@ def main():
running_disc_loss += disc_loss.item()
else:
disc_loss = torch.tensor(0.0)
weighted_d_adversarial_loss = torch.tensor(0.0)
lecam_loss = torch.tensor(0.0)
gradient_penalty_loss = torch.tensor(0.0)
log_step += 1
# Log to tensorboard