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

modifié : dataset/dataset.py

parent 299bf632
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 @ 753d5514
...@@ -254,7 +254,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True): ...@@ -254,7 +254,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
shuffle=shuffle, shuffle=shuffle,
num_workers=8, num_workers=8,
collate_fn=None, collate_fn=None,
pin_memory=False, pin_memory=True,
) )
data_loader_test = data.DataLoader( data_loader_test = data.DataLoader(
dataset=val_dataset, dataset=val_dataset,
...@@ -262,7 +262,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True): ...@@ -262,7 +262,7 @@ def load_data_duo(base_dir, batch_size, args, shuffle=True):
shuffle=shuffle, shuffle=shuffle,
num_workers=8, num_workers=8,
collate_fn=None, collate_fn=None,
pin_memory=False, pin_memory=True,
) )
return data_loader_train, data_loader_test return data_loader_train, data_loader_test
......
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