Skip to content

Commit a599652

Browse files
authored
Update config.py
1 parent b7e9c48 commit a599652

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

config.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ class TrainConfig(BaseConfig):
141141
# number of resblocks for the UNET
142142
net_num_input_res_blocks: int = None
143143
net_enc_num_cls: int = None
144-
num_workers: int = 1
144+
num_workers: int = 0
145145
parallel: bool = False
146146
postfix: str = ''
147147
sample_size: int = 64
@@ -329,7 +329,7 @@ def make_loader(self,
329329
num_workers=num_worker or self.num_workers,
330330
pin_memory=True,
331331
drop_last=drop_last,
332-
multiprocessing_context=get_context('fork'),
332+
multiprocessing_context=None,
333333
)
334334

335335
def make_model_conf(self):

0 commit comments

Comments
 (0)