`Keras.Model.fit`-like torch fit implementation
Project description
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
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torch-model-0.0.1.tar.gz
(2.3 kB
view hashes)
Built Distribution
Close
Hashes for torch_model-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7b47c480cd7bda632b0192ac8a8b1a72d2fa881ad52811f8c90942dcf3a86b8a |
|
MD5 | 7bebbb7dd6a026b9f33239c78571f75a |
|
BLAKE2b-256 | 5048a2125e0c239afe0d6885a7722edbb53bea24e7b3b7c3b659d8b69236aab6 |