File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -339,11 +339,11 @@ def prepare_data_loaders(data_path):
339
339
340
340
data_loader = torch .utils .data .DataLoader (
341
341
dataset , batch_size = train_batch_size ,
342
- sampler = train_sampler , num_workers = 1 )
342
+ sampler = train_sampler )
343
343
344
344
data_loader_test = torch .utils .data .DataLoader (
345
345
dataset_test , batch_size = eval_batch_size ,
346
- sampler = test_sampler , num_workers = 4 , pin_memory = True )
346
+ sampler = test_sampler )
347
347
348
348
return data_loader , data_loader_test
349
349
@@ -455,7 +455,7 @@ def prepare_data_loaders(data_path):
455
455
# quantization parameters in an optimal manner.
456
456
#
457
457
458
- per_channel_quantized_model = load_model (saved_model_dir + float_model_file )
458
+ per_channel_quantized_model = load_model (saved_model_dir + float_model_file )
459
459
per_channel_quantized_model .eval ()
460
460
per_channel_quantized_model .fuse_model ()
461
461
per_channel_quantized_model .qconfig = torch .quantization .get_default_qconfig ('fbgemm' )
You can’t perform that action at this time.
0 commit comments