Skip to content
Snippets Groups Projects
Commit d2dc37e0 authored by Léo Calmettes's avatar Léo Calmettes
Browse files

modifié : dataset/dataset.py

parent 416e304b
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
dataset/dataset.py
+ 2
− 2
View file @ d2dc37e0
...@@ -252,7 +252,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True): ...@@ -252,7 +252,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
dataset=train_dataset, dataset=train_dataset,
batch_size=batch_size, batch_size=batch_size,
shuffle=shuffle, shuffle=shuffle,
num_workers=1, num_workers=8,
collate_fn=None, collate_fn=None,
pin_memory=False, pin_memory=False,
) )
...@@ -260,7 +260,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True): ...@@ -260,7 +260,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
dataset=val_dataset, dataset=val_dataset,
batch_size=batch_size, batch_size=batch_size,
shuffle=shuffle, shuffle=shuffle,
num_workers=1, num_workers=8,
collate_fn=None, collate_fn=None,
pin_memory=False, pin_memory=False,
) )
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment