corentin/torch_utils
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix trainer validation loop

Browse source
This commit is contained in:
Corentin 2021-01-13 00:14:04 +09:00
commit 9ac6fb64e8
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trainer.py
View file

@ -197,7 +197,7 @@ class Trainer:
val_inputs = torch.as_tensor( val_inputs = torch.as_tensor(
self.batch_generator_val.batch_data, dtype=self.data_dtype, device=self.device) self.batch_generator_val.batch_data, dtype=self.data_dtype, device=self.device)
val_labels = torch.as_tensor( val_labels = torch.as_tensor(
self.batch_generator_val.batch_data, dtype=self.label_dtype, device=self.device) self.batch_generator_val.batch_label, dtype=self.label_dtype, device=self.device)
val_pre_process = self.pre_process(val_inputs) val_pre_process = self.pre_process(val_inputs)
val_outputs = self.network(val_pre_process) val_outputs = self.network(val_pre_process)