Commit @8264532c206195d011e8070ce9eae2deed3b4512

Y
List All
Feedback
- This Project
- All Projects

윤영준 2023-06-27

Loss function for Autoencoder

@8264532c206195d011e8070ce9eae2deed3b4512

3f5673b

8264532

model/Discriminator.py

--- model/Discriminator.py

+++ model/Discriminator.py


 
             # Inference function
             ret = self.forward(real_clean)
-            fc_out_o, attention_mask_o, fc2_o = ret["fc_out"], ret["attention_map"] , ret["fc_raw"]
+            fc_out_o, attention_mask_o, fc2_o = ret["fc_out"], ret["attention_map"], ret["fc_raw"]
             ret = self.forward(label_tensor)
-            fc_out_r, attention_mask_r, fc2_r = ret["fc_out"], ret["attention_map"] , ret["fc_raw"]
+            fc_out_r, attention_mask_r, fc2_r = ret["fc_out"], ret["attention_map"], ret["fc_raw"]
 
             l_map = F.mse_loss(attention_map, attention_mask_o) + \
                     F.mse_loss(attention_mask_r, zeros_mask)

Add a comment

Open 0
Closed 0

List

...	...	@@ -31,9 +31,9 @@
31	31
32	32	# Inference function
33	33	ret = self.forward(real_clean)
34		- fc_out_o, attention_mask_o, fc2_o = ret["fc_out"], ret["attention_map"] , ret["fc_raw"]
	34	+ fc_out_o, attention_mask_o, fc2_o = ret["fc_out"], ret["attention_map"], ret["fc_raw"]
35	35	ret = self.forward(label_tensor)
36		- fc_out_r, attention_mask_r, fc2_r = ret["fc_out"], ret["attention_map"] , ret["fc_raw"]
	36	+ fc_out_r, attention_mask_r, fc2_r = ret["fc_out"], ret["attention_map"], ret["fc_raw"]
37	37
38	38	l_map = F.mse_loss(attention_map, attention_mask_o) + \
39	39	F.mse_loss(attention_mask_r, zeros_mask)

Delete comment