Spaces:
Configuration error
Configuration error
Zhikang Niu
commited on
Update trainer.py
Browse files- model/trainer.py +1 -1
model/trainer.py
CHANGED
@@ -172,7 +172,7 @@ class Trainer:
|
|
172 |
train_dataloader = DataLoader(train_dataset, collate_fn=collate_fn, num_workers=num_workers, pin_memory=True,
|
173 |
batch_sampler=batch_sampler)
|
174 |
else:
|
175 |
-
raise ValueError(f"batch_size_type must be either 'sample' or 'frame', but
|
176 |
|
177 |
# accelerator.prepare() dispatches batches to devices;
|
178 |
# which means the length of dataloader calculated before, should consider the number of devices
|
|
|
172 |
train_dataloader = DataLoader(train_dataset, collate_fn=collate_fn, num_workers=num_workers, pin_memory=True,
|
173 |
batch_sampler=batch_sampler)
|
174 |
else:
|
175 |
+
raise ValueError(f"batch_size_type must be either 'sample' or 'frame', but received {self.batch_size_type}")
|
176 |
|
177 |
# accelerator.prepare() dispatches batches to devices;
|
178 |
# which means the length of dataloader calculated before, should consider the number of devices
|