mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

source sync

Browse Source 
PiperOrigin-RevId: 223081582
This commit is contained in:
Roy Frostig 2018-11-27 16:46:45 -08:00
parent 0ea98501aa
commit 14844224fc
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/resnet50.py
View File

@ -125,7 +125,3 @@ if __name__ == "__main__":
opt_state = update(i, opt_state, next(batches))
trained_params = minmax.get_params(opt_state)
if __name__ == '__main__':
from absl import app
app.run(main)