Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def __init__(self, model, im_size = (3,224,224) ):
self.model = model
self.im_size = (1,) + im_size
self.model_ext = FullGradExtractor(model, im_size)
self.biases = self.model_ext.getBiases()
self.checkCompleteness()
def __init__(self, model, im_size = (3,224,224) ):
self.model = model
self.im_size = (1,) + im_size
self.model_ext = FullGradExtractor(model, im_size)