mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-10-24 12:41:49 +08:00
removed self.model refs
This commit is contained in:
1 parent
787f523a71
commit
11b4bc3fbc
1 file changed
+1
-1
@@ -112,12 +112,12 @@ class ExampleModel(RootModule):
|
||||
"""
|
||||
val_loss_mean = 0
|
||||
val_acc_mean = 0
|
||||
pdb.set_trace()
|
||||
for output in outputs:
|
||||
val_loss_mean += output['val_loss']
|
||||
val_acc_mean += output['val_acc']
|
||||
|
||||
val_loss_mean /= len(outputs)
|
||||
val_acc_mean /= len(outputs)
|
||||
tqdm_dic = {'val_loss': val_loss_mean.item(), 'val_acc': val_acc_mean.item()}
|
||||
return tqdm_dic
|
||||
|
||||
|
||||
Reference in new issue
Block a user