opensource/DeOldify
3
0
Fork 0
mirror of https://github.com/jantic/DeOldify.git synced 2026-08-28 09:34:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fixing show_results() functionality

Browse Source
This commit is contained in:
Jason Antic
2023-04-16 12:37:52 -04:00
committed by GitHub
parent 4ff02e5f22
commit e52639d51d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastai/basic_train.py
+1 -1
View File
@@ -347,7 +347,7 @@ class Learner():
if batch is not None: xb,yb = batch
else: xb,yb = self.data.one_batch(ds_type, detach=False, denorm=False)
cb_handler = CallbackHandler(self.callbacks)
#xb,yb = cb_handler.on_batch_begin(xb,yb, train=False)
xb,yb = cb_handler.on_batch_begin(xb,yb, train=False)
if not with_dropout:
preds = loss_batch(self.model.eval(), xb, yb, cb_handler=cb_handler)
else: