implement group-norm - #790
Conversation
![chatgpt-codex-connector[bot]](https://avatars.githubusercontent.com/in/1144995?s=60&v=4){kind=small}
There was a problem hiding this comment.
💡 Codex Review
Here are some automated review suggestions for this pull request.
Reviewed commit: ba3f6ddeec
ℹ️ About Codex in GitHub
Your team has set up Codex to review pull requests in this repo. Reviews are triggered when you
- Open a pull request for review
- Mark a draft as ready
- Comment "@codex review".
If Codex has suggestions, it will comment; otherwise it will react with 👍.
Codex can also answer questions or update the PR. Try commenting "@codex address that feedback".
| BlockType = Literal["conv_next_block", "conv_block"] | ||
| ActivationType = Literal["relu", "gelu", "capped_gelu"] | ||
| NormType = Literal["batch", "instance", "layer"] | ||
| NormType = Literal["batch", "instance", "group", "nonorm", "layer"] |
There was a problem hiding this comment.
When a config sets core_block.block_type: "conv_block" with the newly accepted norm: "group" or "nonorm", validation now succeeds, but the BlockConfig.build conv_block branch still constructs ConvBlock without passing self.norm, and ConvBlock hard-codes BatchNorm2d. That silently runs experiments with batch norm despite the YAML requesting group/no normalization, so either reject these values for conv_block or thread the norm setting through that block too.
Useful? React with 👍 / 👎.
2 participants
implement group and layer normalization with tests. Change norm selection to use build_norm_layer that is more extensible than old if/elif logic.