Closed alexcuozzo-1 closed 3 years ago
In the global SNR pruning mixin, I referenced the model's VDropCentralData as vdrop_data = self.model.vdrop_central_data, but in the original layer-wise version it is referred to as vdrop_data = self.model.module.vdrop_central_data. This would require the model to assign its vdrop to vdrop_central_data. I would like to keep the convention consistent between the two mixins but wasn't sure which to use - any thoughts?
https://ntaresearch.atlassian.net/browse/RES-2264