aws-neuron / aws-neuron-samples

Example code for AWS Neuron SDK developers building inference and training applications
Other
122 stars 33 forks source link

Add CLIP sample script and update VIT sample script #19

Closed aws-tianquaw closed 1 year ago

aws-tianquaw commented 1 year ago

Add CLIP sample script and update VIT sample script.

Verified that with neuron release 2.12, CLIP and VIT training are supported on trn instances.

natemail-aws commented 1 year ago

tested and passed manually. testing in automation