-
- Downloads
Avoid TensorFlow overhead by making one step a batch rather than an
epoch. Avoids memory overhead by only combining up to 100 steps into one epoch, and not changing anything when using only 1 replica (i.e. on CPU).
Showing
- n3fit/src/n3fit/backends/keras_backend/MetaModel.py 33 additions, 2 deletionsn3fit/src/n3fit/backends/keras_backend/MetaModel.py
- n3fit/src/n3fit/backends/keras_backend/callbacks.py 51 additions, 7 deletionsn3fit/src/n3fit/backends/keras_backend/callbacks.py
- n3fit/src/n3fit/stopping.py 3 additions, 3 deletionsn3fit/src/n3fit/stopping.py
Loading
Please register or sign in to comment