Unable to run StabNet Stabilizer

[architmang]

Although we are able to get DUT Stabilizer output & DIFRINT stabilizer output, however we are facing the following error for StabNet Stabilizer:

File "StabNetStabilizer.py", line 36
model.load_state_dict(r_model)
File "/home/user/.conda/envs/DUTCode/lib/python3.6/site-packages/torch/nn/modules/module.py", line 1224, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for stabNet:
Missing key(s) in state_dict: "resnet50.resnet_v2_50_block1_unit_1_bottleneck_v2_preact_FusedBatchNorm.running_mean", "resnet50.resnet_v2_50_block1_unit_1_bottleneck_v2_preact_FusedBatchNorm.running_var", "resnet50.resnet_v2_50_block1_unit_1_bottleneck_v2_conv1_BatchNorm_FusedBatchNorm.running_mean",
......................
"resnet50.resnet_v2_50_postnorm_FusedBatchNorm.running_mean", "resnet50.resnet_v2_50_postnorm_FusedBatchNorm.running_var", "resnet50.resnet_v2_50_logits_Conv2D.weight", "resnet50.resnet_v2_50_logits_Conv2D.bias".

The observation we made is that DUT_stable output has wobble whereas DIFRINT_stable output has artifacts but no wobble. Is the error we are getting somehow causing artifacts in the output?