Closed alvarobartt closed 1 year ago
flaxmodels.resnet.ResNet50
tests/
benchmarks/
ruff
black
examples/
If the above checkbox is checked, describe how you unit-tested it.
Keep the same unit tests but add ResNet50 as a pytest.fixture in order to test safejax on more complex FrozenDict structures.
pytest.fixture
safejax
FrozenDict
✨ Features
flaxmodels.resnet.ResNet50in bothtests/andbenchmarks/ruffandblackover bothexamples/andbenchmarks/too🧪 Tests
If the above checkbox is checked, describe how you unit-tested it.
Keep the same unit tests but add ResNet50 as a
pytest.fixturein order to testsafejaxon more complexFrozenDictstructures.