ARCHIVED, because there is already one for you:

https://github.com/ncullen93/torchsample

---

torch-fit

Keras.Model::fit-like torch fit implementation

Install

$ pip install torch-fit

Usage

from torch_fit import fit

model = MyPyTorchModule()

history = fit(
    model,
    loss='mse',
    metrics='mae',
    epochs=100
)

License

MIT