v0.0.5

patel-zeel released this 29 Nov 11:45

· 4 commits to main since this release

18067c7

What's new?

train_fn now supports dataloader (Torch DataLoader)
predict and predict_class methods from classifiers now support dataloader (Torch DataLoader)
Added alexnet and densenet models

Assets 2