Commit @4480ea0227e7e43134b0098e1c486cb1dc17ae95

Y
List All
Feedback
- This Project
- All Projects

윤영준 2023-07-04

fatal error on weight saving fixed.

@4480ea0227e7e43134b0098e1c486cb1dc17ae95

0f67929

4480ea0

train.py

--- train.py

+++ train.py


 
 resize = torchvision.transforms.Resize((692, 776), antialias=True)
 dataset = ImagePairDataset(clean_data_path, rainy_data_path, transform=resize)
-dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True)
+dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True, num_workers=4)
 # declare generator loss
 
 

 
     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_{epoch}_{day}.pt")
+        torch.save(generator.attentiveRNN.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")
 

Add a comment

Open 0
Closed 0

List

...	...	@@ -71,7 +71,7 @@
71	71
72	72	resize = torchvision.transforms.Resize((692, 776), antialias=True)
73	73	dataset = ImagePairDataset(clean_data_path, rainy_data_path, transform=resize)
74		-dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True)
	74	+dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True, num_workers=4)
75	75	# declare generator loss
76	76
77	77
...	...	@@ -148,7 +148,7 @@
148	148
149	149	day = strftime("%Y-%m-%d %H:%M:%S", gmtime())
150	150	if epoch % save_interval == 0 and epoch != 0:
151		- torch.save(generator.attentionRNN.state_dict(), f"weight/Attention_RNN_{epoch}_{day}.pt")
	151	+ torch.save(generator.attentiveRNN.state_dict(), f"weight/Attention_RNN_{epoch}_{day}.pt")
152	152	torch.save(generator.state_dict(), f"weight/Generator_{epoch}_{day}.pt")
153	153	torch.save(discriminator.state_dict(), f"weight/Discriminator_{epoch}_{day}.pt")
154	154

Delete comment