fix · 4c6497f726 - cvpr - Gitea: Git with a cup of tea

luni/cvpr

fix

This commit is contained in:

luni

2023-02-28 20:04:30 +00:00

parent d5e2c629b7

commit 4c6497f726

1 changed files with 3 additions and 0 deletions

									
										cvae_pyramid.py
									
		+3
		
												View File
												
					@@ -880,6 +880,9 @@ if __name__ == '__main__':

					            train_accumulator.reset()

					            train_accumulator.reset()

					            valid_accumulator.reset()

					            valid_accumulator.reset()

					        if step % 5000 == 0:

					            manager.save(checkpoint_number=step)

					        print(step)

					        print(step)

					        step += 1

					        step += 1