I am trying to decompose the first conv layer of pretrained vgg16 using the package tensorly .and trying to fine tune as follows.
model = models.vgg16(pretrained=True)
layer = model.features._modules[‘0’]
factors=parafac(layer.data, 16, init=‘svd’)
layer = tl.kruskal_to_tensor(factors)
model.features._modules[‘0’] = layer
model = model.to(device)
I am getting the following error
AttributeError: ‘Tensor’ object has no attribute ‘_apply’
Can anyone help?