Batch normalization missing #297
Description
I noticed that batch normalization is missing in the MLP and ShapedMLP backbone, we should probably add it to both implementations.
Metadata
Metadata
Assignees
Labels
No labels
{{ message }}
I noticed that batch normalization is missing in the MLP and ShapedMLP backbone, we should probably add it to both implementations.