We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 70b7691 commit be8fc1fCopy full SHA for be8fc1f
the_well/benchmark/trainer/training.py
@@ -314,7 +314,11 @@ def validation_loop(
314
loss_dict = {}
315
time_logs = {}
316
count = 0
317
- denom = len(dataloader) if full else min(self.short_validation_length, len(dataloader))
+ denom = (
318
+ len(dataloader)
319
+ if full
320
+ else min(self.short_validation_length, len(dataloader))
321
+ )
322
with torch.autocast(
323
self.device.type, enabled=self.enable_amp, dtype=self.amp_type
324
):
0 commit comments