torch.nn.modules.module.ModuleAttributeError: 'BatchNorm' object has no attribute 'weight'

[cravisjan97]

I am training your UNet mesh segmenter model by running training_job_example_aliens_unet.py file and I get this error when I am trying to initialize the model. Is there a fix for this? Can I disable initialization and expect the same accuracy?

[HaominJiang]

I am training your UNet mesh segmenter model by running training_job_example_aliens_unet.py file and I get this error when I am trying to initialize the model. Is there a fix for this? Can I disable initialization and expect the same accuracy?

Add statement hasattr(module, 'weight') before class_name.find('BatchNorm') != -1 in

PD-MeshNet/pd_mesh_net/utils/models.py

Line 123 in e3f6c01

elif (class_name.find('BatchNorm') != -1):

works for me. It could be changed to

elif hasattr(module, 'weight') and class_name.find('BatchNorm') != -1

Not sure if such changes will destroy the initialization of the network.