Skip to content
Snippets Groups Projects
Commit 95d16f1d authored by Alexandre Chapin's avatar Alexandre Chapin :race_car:
Browse files

Right function for dataloaders

parent 65810114
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
train_lit.py
+ 1
− 1
View file @ 95d16f1d
...@@ -259,7 +259,7 @@ def main(cfg) -> None: ...@@ -259,7 +259,7 @@ def main(cfg) -> None:
val_loader = DataLoader(val_dataset, batch_size=batch_size, num_workers=num_workers, shuffle=True) val_loader = DataLoader(val_dataset, batch_size=batch_size, num_workers=num_workers, shuffle=True)
test_loader = DataLoader(test_dataset, batch_size=batch_size, num_workers=num_workers) test_loader = DataLoader(test_dataset, batch_size=batch_size, num_workers=num_workers)
train_loader, val_loader, test_loader = fabric._setup_dataloader(train_loader, val_loader, test_loader) train_loader, val_loader, test_loader = fabric.setup_dataloaders(train_loader, val_loader, test_loader)
vis_loader_val = DataLoader(val_dataset, batch_size=12, num_workers=num_workers) vis_loader_val = DataLoader(val_dataset, batch_size=12, num_workers=num_workers)
data_vis_val = next(iter(vis_loader_val)) # Validation set data for visualization data_vis_val = next(iter(vis_loader_val)) # Validation set data for visualization
......
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