wehub-resource-sync
22 lines
708 B
Python
22 lines
708 B
Python
"""Various callbacks to customize get_preds behaviors
|
|
|
|
Docs: https://docs.fast.ai/callback.preds.html.md"""
|
|
|
|
# AUTOGENERATED! DO NOT EDIT! File to edit: ../../nbs/18b_callback.preds.ipynb.
|
|
|
|
# %% auto #0
|
|
__all__ = ['MCDropoutCallback']
|
|
|
|
# %% ../../nbs/18b_callback.preds.ipynb #baa1e5e1
|
|
from ..basics import *
|
|
|
|
# %% ../../nbs/18b_callback.preds.ipynb #8f0bcd2d
|
|
class MCDropoutCallback(Callback):
|
|
def before_validate(self):
|
|
for m in [m for m in flatten_model(self.model) if 'dropout' in m.__class__.__name__.lower()]:
|
|
m.train()
|
|
|
|
def after_validate(self):
|
|
for m in [m for m in flatten_model(self.model) if 'dropout' in m.__class__.__name__.lower()]:
|
|
m.eval()
|