Visitar URL original
Set dataloader_workers to 0 · DeepLabCut/DeepLabCut@e778557 · GitHub
Skip to content

Commit e778557

Browse files
committed
Set dataloader_workers to 0
1 parent 196bf9a commit e778557

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

‎examples/testscript_pytorch_multi_animal.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -67,12 +67,14 @@ def main(
6767
"train_settings.display_iters": 50,
6868
"train_settings.epochs": epochs_,
6969
"train_settings.batch_size": batch_size,
70+
"train_settings.dataloader_workers": 0,
7071
"runner.device": device,
7172
"runner.snapshots.save_epochs": save_epochs,
7273
"runner.snapshots.max_snapshots": max_snapshots_to_keep,
7374
"detector.train_settings.display_iters": 1,
7475
"detector.train_settings.epochs": detector_epochs,
7576
"detector.train_settings.batch_size": detector_batch_size,
77+
"detector.train_settings.dataloader_workers": 0,
7678
"detector.runner.snapshots.save_epochs": save_epochs,
7779
"detector.runner.snapshots.max_snapshots": max_snapshots_to_keep,
7880
"logger": logger,

0 commit comments

Comments
 (0)