Skip to content

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