google-research / google-research/sam
Regarding mentioning a TF2 implementation
Open
- Dominant language
- Python
- Stars
- 646
- Forks
- 81
- PR merge metrics
- No merged PRs in 30d
Description
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](https://github.com/sayakpaul/Sharpness-Aware-Minimization-TensorFlow). 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.
Contributor guide
Assessment
This issue has not been assessed yet.