def append_parameter_magnitudes()

in train_deep_sdf.py [0:0]

• 7 lines of code
• 5 McCabe index (conditional complexity)

def append_parameter_magnitudes(param_mag_log, model):
    for name, param in model.named_parameters():
        if len(name) > 7 and name[:7] == "module.":
            name = name[7:]
        if name not in param_mag_log.keys():
            param_mag_log[name] = []
        param_mag_log[name].append(param.data.norm().item())