mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00

Files

Ean Garvey a07fe80530 Update OPT, ResNet example scripts. (#1492 )

* Update API in OPT example.

* fix resnet50 script

* Add OPT1.3b test script.

2023-06-05 20:19:35 -07:00

opt_causallm_torch_test.py

Update OPT, ResNet example scripts. (#1492 )

2023-06-05 20:19:35 -07:00

opt_torch_test.py

Update OPT, ResNet example scripts. (#1492 )

2023-06-05 20:19:35 -07:00

README.md

Cleanup tank directory and move instructions to tank/README.md (#401 )

2022-10-13 12:20:02 -05:00

shark_hf_opt.py

Update OPT, ResNet example scripts. (#1492 )

2023-06-05 20:19:35 -07:00

README.md

Running Different OPT Variants

To run different sizes of OPT, change the string OPT_MODEL string in opt_torch_test.py. The default is 350m parameters. 66b cases also exist in the file, simply uncomment the test cases.