Commit @5304dc89ebf36df5bc303c1f53efa998a6d4b57b

Y
List All
Feedback
- This Project
- All Projects

윤영준 2023-06-30

working on parser

@5304dc89ebf36df5bc303c1f53efa998a6d4b57b

ea5d2cb

5304dc8

train.py

--- train.py

+++ train.py


 
 # declare generator loss
 
-optimizer_G = torch.optim.Adam(generator.parameters(), lr=lr)
-optimizer_D = torch.optim.Adam(generator.parameters(), lr=lr)
+optimizer_G = torch.optim.Adam(generator.parameters(), lr=generator_learning_rate)
+optimizer_D = torch.optim.Adam(generator.parameters(), lr=discriminator_learning_rate)
 
 for epoch_num, epoch in enumerate(range(epochs)):
     for i, (imgs, _) in enumerate(dataloader):

Add a comment

Open 0
Closed 0

List

...	...	@@ -60,8 +60,8 @@
60	60
61	61	# declare generator loss
62	62
63		-optimizer_G = torch.optim.Adam(generator.parameters(), lr=lr)
64		-optimizer_D = torch.optim.Adam(generator.parameters(), lr=lr)
	63	+optimizer_G = torch.optim.Adam(generator.parameters(), lr=generator_learning_rate)
	64	+optimizer_D = torch.optim.Adam(generator.parameters(), lr=discriminator_learning_rate)
65	65
66	66	for epoch_num, epoch in enumerate(range(epochs)):
67	67	for i, (imgs, _) in enumerate(dataloader):

Delete comment