Visitar URL original
Print num workers · DeepLabCut/DeepLabCut@5a4df95 · GitHub
Skip to content

Commit 5a4df95

Browse files
maximpavlivmaximpavliv
committed
Print num workers
1 parent 196bf9a commit 5a4df95

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

‎deeplabcut/pose_estimation_pytorch/apis/training.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -156,6 +156,7 @@ def train(
156156

157157
batch_size = run_config["train_settings"]["batch_size"]
158158
num_workers = run_config["train_settings"]["dataloader_workers"]
159+
print(f"num_workers: {num_workers}")
159160
pin_memory = run_config["train_settings"]["dataloader_pin_memory"]
160161
train_dataloader = DataLoader(
161162
train_dataset,

0 commit comments

Comments
 (0)