Base
Base Class of Model API
eval(self)
Set learner to eval mode for validation
prepare_model(self, learner)
Inplace ops for preparing model via HF Accelerator. Automatically sends to device.
train(self)
Set learner to training mode
Last update: October 3, 2021