Base

Bases: Base

Base Class of Model API implemented with HF Accelerate

backward(loss)

model.backward(loss)

eval()

Set learner to eval mode for validation

save(path, save_extra=False)

save model

train()

Set learner to training mode

---

Last update: October 3, 2021