Zhikang Niu commited on
Commit
edc189f
·
unverified ·
1 Parent(s): 77abf4e

Update trainer.py

Browse files
Files changed (1) hide show
  1. 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 recieved {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
 
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