Base

Base Class of Model API implemented with HF Accelerate

backward(self, loss)

model.backward(loss)

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.

save(self, path, save_extra=True)

save model

train(self)

Set learner to training mode

---

Last update: October 3, 2021