search

tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.
Apache License 2.0
471 stars 74 forks source link

Diet: Enable TTNN support #9774

Open Sudharsan-V opened 4 months ago

Sudharsan-V commented 4 months ago

Enable support for Diet model using TTNN ops.

kkeerthana0573 commented 4 months ago

Reference and ttnn implementation for DeiT model is completed. Demo for DeiTForImageClassificationWithTeacher is completed. Corresponding commit : commit Corresponding PR : #10344 Sample outputs for DeiTForImageClassificationWithTeacher:

Expected Label: stinkhorn, carrion fungus
ttnn Prediction: stinkhorn, carrion fungus
Torch Prediction: stinkhorn, carrion fungus
Expected Label: vine snake
ttnn Prediction: vine snake
Torch Prediction: vine snake

Metrics Used:

Accuracy for 200 input samples from Imagenet-1k Validation Dataset: 0.8

Results are recorded in the sheet.