306 Commits

Author SHA1 Message Date
Prashant Kumar
db5be15310 Centralize the parser location. Also add the --device flag.
Centralized the shark_args parser. Also added the --device flag that
specifies the device on which the inference or training is to be done.
2022-05-02 15:35:22 +05:30
Prashant Kumar
bd212634c1 Add CI for testing of various models. (#28)
Added pytest CI for testing for Pytorch Inference and vision models.
2022-04-27 06:52:49 -07:00
Prashant Kumar
3ed5a9c84e Change the python import of shark. 2022-04-26 20:55:33 +05:30
Prashant Kumar
37468f7bb8 Enable Shark Trainer APIs 2022-04-21 20:31:30 +05:30
anush elangovan
e0005b6335 Add github python gitignore 2022-04-13 13:24:38 -07:00
Prashant Kumar
f9864b4ce1 Modify the directory structure to remove conflicts.
The directory structure is modified, also disabled the aot_module
unless the shark_trainer is deployed.
2022-04-04 21:23:33 +05:30