Commit @50cf09cb7dbe31caabcfece6c9df2003156e7b5e

Y
List All
Feedback
- This Project
- All Projects

윤영준 2023-07-04

minor fixes

@50cf09cb7dbe31caabcfece6c9df2003156e7b5e

0e2104d

50cf09c

train.py

--- train.py

+++ train.py


 
     day = strftime("%Y-%m-%d %H:%M:%S", gmtime())
     if epoch % save_interval == 0 and epoch != 0:
-        torch.save(generator.attentionRNN.state_dict(), f"weight/Attention_RNN_{day}.pt")
-        torch.save(generator.state_dict(), f"weight/Generator_{day}.pt")
-        torch.save(discriminator.state_dict(), f"weight/Discriminator_{day}.pt")
+        torch.save(generator.attentionRNN.state_dict(), f"weight/Attention_RNN_{epoch}_{day}.pt")
+        torch.save(generator.state_dict(), f"weight/Generator_{epoch}_{day}.pt")
+        torch.save(discriminator.state_dict(), f"weight/Discriminator_{epoch}_{day}.pt")
 
 server_process.terminate()
-
-
-
-
 
 
 ## RNN 따로 돌리고 CPU로 메모리 옳기고

Add a comment

Open 0
Closed 0

List

...	...	@@ -147,15 +147,11 @@
147	147
148	148	day = strftime("%Y-%m-%d %H:%M:%S", gmtime())
149	149	if epoch % save_interval == 0 and epoch != 0:
150		- torch.save(generator.attentionRNN.state_dict(), f"weight/Attention_RNN_{day}.pt")
151		- torch.save(generator.state_dict(), f"weight/Generator_{day}.pt")
152		- torch.save(discriminator.state_dict(), f"weight/Discriminator_{day}.pt")
	150	+ torch.save(generator.attentionRNN.state_dict(), f"weight/Attention_RNN_{epoch}_{day}.pt")
	151	+ torch.save(generator.state_dict(), f"weight/Generator_{epoch}_{day}.pt")
	152	+ torch.save(discriminator.state_dict(), f"weight/Discriminator_{epoch}_{day}.pt")
153	153
154	154	server_process.terminate()
155		-
156		-
157		-
158		-
159	155
160	156
161	157	## RNN 따로 돌리고 CPU로 메모리 옳기고

Delete comment