Update train_ms.py · bd4591aa2c - sbv2-v2 - Neodyland Git Server

tuna2134/sbv2-v2

Update train_ms.py

This commit is contained in:

Sihan Wang

2023-12-20 02:56:32 +08:00

parent 2e7c173de2

commit bd4591aa2c

1 changed files with 1 additions and 0 deletions

									
										1

train_ms.py
									
												View File
												
					@@ -387,6 +387,7 @@ def run():

					            )

					            )

					        scheduler_g.step()

					        scheduler_g.step()

					        scheduler_d.step()

					        scheduler_d.step()

					        scheduler_wd.step()

					        if net_dur_disc is not None:

					        if net_dur_disc is not None:

					            scheduler_dur_disc.step()

					            scheduler_dur_disc.step()