Unit Test Draft on Model Type

eto-ai / rikai

Parquet-based ML data format optimized for working with unstructured data

https://rikai.readthedocs.io/en/latest/

Apache License 2.0

137 stars 19 forks source link

Closed da-tubi closed 2 years ago

da-tubi commented 2 years ago

This is a draft PR waiting for @eddyxu 's feedback.

This first three commits are from #542 , the last two commits belong to this PR.

How to run the unit test on Model Type:

cd experimental/torchhub/
pytest -s -o log_cli=true tests/test_resnet.py -k test_resnet_model_type

da-liii commented 2 years ago

will create another pr rebased on main