Error of missing argument 'extra_norm' when trainning #4

Hi,

First off thanks for the interesting paper and accompanying code!
I'm trying to execute the training code, and I'm having trouble missing argument extra_norm in line 197-198 function forward() of models/networks.py. I found forward() in line 70 (class StructuralDescriptor) of models/layers/mesh_net.py needed to parse arg extra_norm but it's missing in models/networks.py.

Kindly help.