(module, name, submodule)
| 201 | if include_bn_running_stats: |
| 202 | # Helper to add submodule specified as a named_module |
| 203 | def _add_submodule(module, name, submodule): |
| 204 | split = name.rsplit('.', 1) |
| 205 | if len(split) > 1: |
| 206 | module.get_submodule(split[0]).add_module(split[1], submodule) |
| 207 | else: |
| 208 | module.add_module(name, submodule) |
| 209 | |
| 210 | # Freeze batch norm |
| 211 | if mode == 'freeze': |