Commit @e6ed062ebe53e8e174b55c4cb3d5d020fd1c12a4

Y
List All
Feedback
- This Project
- All Projects

윤영준 2023-07-05

Generator visualizer

@e6ed062ebe53e8e174b55c4cb3d5d020fd1c12a4

7cbe1b2

e6ed062

train.py

--- train.py

+++ train.py


         vis.line(Y=np.array([discriminator_loss.item()]), X=np.array([epoch * epoch_num + i]), win=Discriminator_loss_window,
                  update='append')
         vis.image(generator_attention_map[-1][0,0,:,:], win=Attention_map_visualizer, opts=dict(title="Attention Map"))
-        vis.image(generator_result['skip_3'][-1][0,0,:,:], win=Generator_output_visualizer, opts=dict(title="Generator Output"))
+        vis.image(generator_result[-1], win=Generator_output_visualizer, opts=dict(title="Generator Output"))
     day = strftime("%Y-%m-%d %H:%M:%S", gmtime())
     if epoch % save_interval == 0 and epoch != 0:
         torch.save(generator.attentiveRNN.state_dict(), f"weight/Attention_RNN_{epoch}_{day}.pt")

Add a comment

Open 0
Closed 0

List

...	...	@@ -154,7 +154,7 @@
154	154	vis.line(Y=np.array([discriminator_loss.item()]), X=np.array([epoch * epoch_num + i]), win=Discriminator_loss_window,
155	155	update='append')
156	156	vis.image(generator_attention_map[-1][0,0,:,:], win=Attention_map_visualizer, opts=dict(title="Attention Map"))
157		- vis.image(generator_result['skip_3'][-1][0,0,:,:], win=Generator_output_visualizer, opts=dict(title="Generator Output"))
	157	+ vis.image(generator_result[-1], win=Generator_output_visualizer, opts=dict(title="Generator Output"))
158	158	day = strftime("%Y-%m-%d %H:%M:%S", gmtime())
159	159	if epoch % save_interval == 0 and epoch != 0:
160	160	torch.save(generator.attentiveRNN.state_dict(), f"weight/Attention_RNN_{epoch}_{day}.pt")

Delete comment