heat
heat copied to clipboard
batch normalization
Implementation of a parallel version of Batch Normalisation for Model and Data parallization
@krajsek can we break this in smaller tasks?
we could divide it into model and data parallel BN. Ich have already started some code with data parallelism, have to check how far it is.
@krajsek are there news within this issue?
(Review within #1109)
@mrfh92 The data-parallel version works, but still needs some more testing: https://github.com/helmholtz-analytics/heat/blob/features/589-batch-normalization-3/heat/nn/batch_norm.py
ok sounds cool ... then we leave this definitely open :+1: (Reviewed within #1109)
Just to keep the overview: This is actually https://github.com/helmholtz-analytics/heat/pull/748?