Commit @0e2104de24b5e5f308b1cb2fa2979bd39ca9ef15

Y
List All
Feedback
- This Project
- All Projects

윤영준 2023-07-04

fixed a mistake of putting server termination code before excuting and declaring server process

@0e2104de24b5e5f308b1cb2fa2979bd39ca9ef15

9773fd6

0e2104d

train.py

--- train.py

+++ train.py


 from tools.dataloader import ImagePairDataset
 
 
-
-
-
 # this function is from https://github.com/eriklindernoren/PyTorch-GAN/blob/master/implementations/dualgan/models.py
 # MIT license
 def weights_init_normal(m):

 
 # ------visdom visualizer ----------
 server_process = subprocess.Popen("python -m visdom.server", shell=True)
+# to ensure the visdom server process must stop whenever the script is terminated.
 def cleanup():
     server_process.terminate()
 atexit.register(cleanup)

Add a comment

Open 0
Closed 0

List

...	...	@@ -22,9 +22,6 @@
22	22	from tools.dataloader import ImagePairDataset
23	23
24	24
25		-
26		-
27		-
28	25	# this function is from https://github.com/eriklindernoren/PyTorch-GAN/blob/master/implementations/dualgan/models.py
29	26	# MIT license
30	27	def weights_init_normal(m):
...	...	@@ -83,6 +80,7 @@
83	80
84	81	# ------visdom visualizer ----------
85	82	server_process = subprocess.Popen("python -m visdom.server", shell=True)
	83	+# to ensure the visdom server process must stop whenever the script is terminated.
86	84	def cleanup():
87	85	server_process.terminate()
88	86	atexit.register(cleanup)

Delete comment