torch-model 0.0.1

`Keras.Model.fit`-like torch fit implementation

torch-model

Keras-like torch module wrapper

Tested with Torch 1.8.0

Install

$ pip install torch-model

Usage

from torch_model import Model
from torch_model.callbacks import EarlyStopping

model = Model(MyPyTorchModule())

model.compile(
    loss='mse',
    optimizer='adam',
    metrics=['mae', 'mape']
)

history = model.fit(
    train_ds,
    validation_data=val_ds,
    callbacks=[
        EarlyStopping(monitor='val_loss', patience=5)
    ]
)

# The original torch module
model.module

License

MIT