Commit @0f679299e8739a05d88292e8a860bd9fdf5305c9

윤영준 2023-07-04

added num_worker for argparser

@0f679299e8739a05d88292e8a860bd9fdf5305c9

50cf09c

0f67929

tools/argparser.py

--- tools/argparser.py

+++ tools/argparser.py


     parser.add_argument("--batch_size", "-b", type=int, required=True, help="Size of single batch")
     parser.add_argument("--save_interval", "-s", type=int, required=True, help="Interval for saving weights")
     parser.add_argument("--sample_interval", type=int, required=True, help="Interval for saving inference result")
+    parser.add_argument("--num_worker", "-j", type=int, default=4, help="Dataloader's number of threads")
+    # its j because make uses -j for number of threads
     parser.add_argument("--device", "-d", type=str, default="cpu", choices=["cpu", "cuda"], help="Device to use for "
                                                                                                  "computation")
     parser.add_argument("--load", "-l", type=str, default=None, help="Path to previous weights for continuing training")

50cf09c

0f67929

train.py

--- train.py

+++ train.py


 batch_size = args.batch_size
 save_interval = args.save_interval
 sample_interval = args.sample_interval
+num_worker = args.num_worker
 device = args.device
 load = args.load
 generator_learning_rate = args.generator_learning_rate

Add a comment

Open 0
Closed 0

List

...	...	@@ -8,6 +8,8 @@
8	8	parser.add_argument("--batch_size", "-b", type=int, required=True, help="Size of single batch")
9	9	parser.add_argument("--save_interval", "-s", type=int, required=True, help="Interval for saving weights")
10	10	parser.add_argument("--sample_interval", type=int, required=True, help="Interval for saving inference result")
	11	+ parser.add_argument("--num_worker", "-j", type=int, default=4, help="Dataloader's number of threads")
	12	+ # its j because make uses -j for number of threads
11	13	parser.add_argument("--device", "-d", type=str, default="cpu", choices=["cpu", "cuda"], help="Device to use for "
12	14	"computation")
13	15	parser.add_argument("--load", "-l", type=str, default=None, help="Path to previous weights for continuing training")

...	...	@@ -40,6 +40,7 @@
40	40	batch_size = args.batch_size
41	41	save_interval = args.save_interval
42	42	sample_interval = args.sample_interval
	43	+num_worker = args.num_worker
43	44	device = args.device
44	45	load = args.load
45	46	generator_learning_rate = args.generator_learning_rate

Delete comment