optax icon indicating copy to clipboard operation
optax copied to clipboard

Published 20 hours ago • verified publisher icon google-deepmind

Remove TF dependency in adversarial_training example

Open mmhamdy opened this issue 1 year ago • 2 comments

This PR removes Tensorflow as a dependency in adversarial_training.ipynb, and uses only TFDS, with grain for loading and preparing the dataset.

Addressing issue #656

mmhamdy avatar Feb 07 '24 11:02 mmhamdy

@mmhamdy : now that grain has builds for 3.9 (https://pypi.org/project/grain-nightly/0.0.6/#files), can you update with master to see if the tests pass?

fabianp avatar Feb 20 '24 09:02 fabianp

Great! But I think we need to start with #768 in which I've added grain as doc dependency. I've updated it and it has passed all tests successfully 🚀

mmhamdy avatar Feb 20 '24 23:02 mmhamdy