Regarding mentioning a TF2 implementation
sayakpaul opened this issue · comments
Hi @PForet. Thank you for open-sourcing SAM.
I have been dabbling with SAM for the past few days and I am pleased to share a minimal TF2 implementation of SAM. The repo includes a Colab Notebook demoing SAM on CIFAR10 with TPUv2-8 hardware.
Is it possible to mention this from the README? It might help the community to quickly experiment with SAM and expand on it.