You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
failed to find NodeArg by name f2_bn2.running_var_grad
the BatchNorm training inputs (running mean and running var) are not trainable. They are initializers in the model that get updated, but they are not trainable.
You will have to put the initializers in the frozen_params list while generating the training artifacts:
frozen_params= ["f2_bn2.running_var", "f2_bn2.running_mean"] # and potentially other similar initializers
Describe the issue
When trying to get the training model with a batchnormalisation node I get the error:
failed to find NodeArg by name f2_bn2.running_var_grad
Is batchnormalisation not supported for training?
This is a trained model converted from here: https://github.com/suragnair/alpha-zero-general (othello)
To reproduce
Create a training artefact with a batch normalisation node.
Urgency
No response
ONNX Runtime Installation
Released Package
ONNX Runtime Version or Commit ID
1.15.1
PyTorch Version
1.13.1
Execution Provider
CUDA
Execution Provider Library Version
CUDA 1.13.1
The text was updated successfully, but these errors were encountered: