frgfm / holocron Goto Github PK

File "/path/to/poly_loss.py", in forward
    return poly_loss(x, target, self.eps, self.weight, self.ignore_index, self.reduction)
File "/path/to/poly_loss.py", in poly_loss
    logpt = logpt.transpose(1, 0).flatten(1).gather(0, target.view(1, -1)).squeeze()
RuntimeError: gather(): Expected dtype int64 for index

pretrained = True
backbone= "resnet50"
model = holocron.models.__dict__[backbone](
       pretrained, num_classes=1)

Missing key(s) in state_dict: "features.0.weight", "features.1.weight", "features.1.bias", "features.1.running_mean", "features.1.running_var", "features.4.0.conv.0.weight", "features.4.0.conv.1.weight", "features.4.0.conv.1.bias", "features.4.0.conv.1.running_mean", "features.4.0.conv.1.running_var", "features.4.0.conv.3.weight", "features.4.0.conv.4.weight", "features.4.0.conv.4.bias", "features.4.0.conv.4.running_mean", "features.4.0.conv.4.running_var", "features.4.0.conv.6.weight", "features.4.0.conv.7.weight", "features.4.0.conv.7.bias", "features.4.0.conv.7.running_mean", "features.4.0.conv.7.running_var", "features.4.0.downsample.0.weight", "features.4.0.downsample.1.weight", "features.4.0.downsample.1.bias", "features.4.0.downsample.1.running_mean", "features.4.0.downsample.1.running_var", "features.4.1.conv.0.weight", "features.4.1.conv.1.weight", "features.4.1.conv.1.bias", "features.4.1.conv.1.running_mean", "features.4.1.conv.1.running_var", "features.4.1.conv.3.weight", "features.4.1.conv.4.weight", "features.4.1.conv.4.bias", "features.4.1.conv.4.running_mean", "features.4.1.conv.4.running_var", "features.4.1.conv.6.weight", "features.4.1.conv.7.weight", "features.4.1.conv.7.bias", "features.4.1.conv.7.running_mean", "features.4.1.conv.7.running_var", "features.4.2.conv.0.weight", "features.4.2.conv.1.weight", "features.4.2.conv.1.bias", "features.4.2.conv.1.running_mean", "features.4.2.conv.1.running_var", "features.4.2.conv.3.weight", "features.4.2.conv.4.weight", "features.4.2.conv.4.bias", "features.4.2.conv.4.running_mean", "features.4.2.conv.4.running_var", "features.4.2.conv.6.weight", "features.4.2.conv.7.weight", "features.4.2.conv.7.bias", "features.4.2.conv.7.running_mean", "features.4.2.conv.7.running_var", "features.5.0.conv.0.weight", "features.5.0.conv.1.weight", "features.5.0.conv.1.bias", "features.5.0.conv.1.running_mean", "features.5.0.conv.1.running_var", "features.5.0.conv.3.weight", "features.5.0.conv.4.weight", "features.5.0.conv.4.bias", "features.5.0.conv.4.running_mean", "features.5.0.conv.4.running_var", "features.5.0.conv.6.weight", "features.5.0.conv.7.weight", "features.5.0.conv.7.bias", "features.5.0.conv.7.running_mean", "features.5.0.conv.7.running_var", "features.5.0.downsample.0.weight", "features.5.0.downsample.1.weight", "features.5.0.downsample.1.bias", "features.5.0.downsample.1.running_mean", "features.5.0.downsample.1.running_var", "features.5.1.conv.0.weight", "features.5.1.conv.1.weight", "features.5.1.conv.1.bias", "features.5.1.conv.1.running_mean", "features.5.1.conv.1.running_var", "features.5.1.conv.3.weight", "features.5.1.conv.4.weight", "features.5.1.conv.4.bias", "features.5.1.conv.4.running_mean", "features.5.1.conv.4.running_var", "features.5.1.conv.6.weight", "features.5.1.conv.7.weight", "features.5.1.conv.7.bias", "features.5.1.conv.7.running_mean", "features.5.1.conv.7.running_var", "features.5.2.conv.0.weight", "features.5.2.conv.1.weight", "features.5.2.conv.1.bias", "features.5.2.conv.1.running_mean", "features.5.2.conv.1.running_var", "features.5.2.conv.3.weight", "features.5.2.conv.4.weight", "features.5.2.conv.4.bias", "features.5.2.conv.4.running_mean", "features.5.2.conv.4.running_var", "features.5.2.conv.6.weight", "features.5.2.conv.7.weight", "features.5.2.conv.7.bias", "features.5.2.conv.7.running_mean", "features.5.2.conv.7.running_var", "features.5.3.conv.0.weight", "features.5.3.conv.1.weight", "features.5.3.conv.1.bias", "features.5.3.conv.1.running_mean", "features.5.3.conv.1.running_var", "features.5.3.conv.3.weight", "features.5.3.conv.4.weight", "features.5.3.conv.4.bias", "features.5.3.conv.4.running_mean", "features.5.3.conv.4.running_var", "features.5.3.conv.6.weight", "features.5.3.conv.7.weight", "features.5.3.conv.7.bias", "features.5.3.conv.7.running_mean", "features.5.3.conv.7.running_var", "features.6.0.conv.0.weight", "features.6.0.conv.1.weight", "features.6.0.conv.1.bias", "features.6.0.conv.1.running_mean", "features.6.0.conv.1.running_var", "features.6.0.conv.3.weight", "features.6.0.conv.4.weight", "features.6.0.conv.4.bias", "features.6.0.conv.4.running_mean", "features.6.0.conv.4.running_var", "features.6.0.conv.6.weight", "features.6.0.conv.7.weight", "features.6.0.conv.7.bias", "features.6.0.conv.7.running_mean", "features.6.0.conv.7.running_var", "features.6.0.downsample.0.weight", "features.6.0.downsample.1.weight", "features.6.0.downsample.1.bias", "features.6.0.downsample.1.running_mean", "features.6.0.downsample.1.running_var", "features.6.1.conv.0.weight", "features.6.1.conv.1.weight", "features.6.1.conv.1.bias", "features.6.1.conv.1.running_mean", "features.6.1.conv.1.running_var", "features.6.1.conv.3.weight", "features.6.1.conv.4.weight", "features.6.1.conv.4.bias", "features.6.1.conv.4.running_mean", "features.6.1.conv.4.running_var", "features.6.1.conv.6.weight", "features.6.1.conv.7.weight", "features.6.1.conv.7.bias", "features.6.1.conv.7.running_mean", "features.6.1.conv.7.running_var", "features.6.2.conv.0.weight", "features.6.2.conv.1.weight", "features.6.2.conv.1.bias", "features.6.2.conv.1.running_mean", "features.6.2.conv.1.running_var", "features.6.2.conv.3.weight", "features.6.2.conv.4.weight", "features.6.2.conv.4.bias", "features.6.2.conv.4.running_mean", "features.6.2.conv.4.running_var", "features.6.2.conv.6.weight", "features.6.2.conv.7.weight", "features.6.2.conv.7.bias", "features.6.2.conv.7.running_mean", "features.6.2.conv.7.running_var", "features.6.3.conv.0.weight", "features.6.3.conv.1.weight", "features.6.3.conv.1.bias", "features.6.3.conv.1.running_mean", "features.6.3.conv.1.running_var", "features.6.3.conv.3.weight", "features.6.3.conv.4.weight", "features.6.3.conv.4.bias", "features.6.3.conv.4.running_mean", "features.6.3.conv.4.running_var", "features.6.3.conv.6.weight", "features.6.3.conv.7.weight", "features.6.3.conv.7.bias", "features.6.3.conv.7.running_mean", "features.6.3.conv.7.running_var", "features.6.4.conv.0.weight", "features.6.4.conv.1.weight", "features.6.4.conv.1.bias", "features.6.4.conv.1.running_mean", "features.6.4.conv.1.running_var", "features.6.4.conv.3.weight", "features.6.4.conv.4.weight", "features.6.4.conv.4.bias", "features.6.4.conv.4.running_mean", "features.6.4.conv.4.running_var", "features.6.4.conv.6.weight", "features.6.4.conv.7.weight", "features.6.4.conv.7.bias", "features.6.4.conv.7.running_mean", "features.6.4.conv.7.running_var", "features.6.5.conv.0.weight", "features.6.5.conv.1.weight", "features.6.5.conv.1.bias", "features.6.5.conv.1.running_mean", "features.6.5.conv.1.running_var", "features.6.5.conv.3.weight", "features.6.5.conv.4.weight", "features.6.5.conv.4.bias", "features.6.5.conv.4.running_mean", "features.6.5.conv.4.running_var", "features.6.5.conv.6.weight", "features.6.5.conv.7.weight", "features.6.5.conv.7.bias", "features.6.5.conv.7.running_mean", "features.6.5.conv.7.running_var", "features.7.0.conv.0.weight", "features.7.0.conv.1.weight", "features.7.0.conv.1.bias", "features.7.0.conv.1.running_mean", "features.7.0.conv.1.running_var", "features.7.0.conv.3.weight", "features.7.0.conv.4.weight", "features.7.0.conv.4.bias", "features.7.0.conv.4.running_mean", "features.7.0.conv.4.running_var", "features.7.0.conv.6.weight", "features.7.0.conv.7.weight", "features.7.0.conv.7.bias", "features.7.0.conv.7.running_mean", "features.7.0.conv.7.running_var", "features.7.0.downsample.0.weight", "features.7.0.downsample.1.weight", "features.7.0.downsample.1.bias", "features.7.0.downsample.1.running_mean", "features.7.0.downsample.1.running_var", "features.7.1.conv.0.weight", "features.7.1.conv.1.weight", "features.7.1.conv.1.bias", "features.7.1.conv.1.running_mean", "features.7.1.conv.1.running_var", "features.7.1.conv.3.weight", "features.7.1.conv.4.weight", "features.7.1.conv.4.bias", "features.7.1.conv.4.running_mean", "features.7.1.conv.4.running_var", "features.7.1.conv.6.weight", "features.7.1.conv.7.weight", "features.7.1.conv.7.bias", "features.7.1.conv.7.running_mean", "features.7.1.conv.7.running_var", "features.7.2.conv.0.weight", "features.7.2.conv.1.weight", "features.7.2.conv.1.bias", "features.7.2.conv.1.running_mean", "features.7.2.conv.1.running_var", "features.7.2.conv.3.weight", "features.7.2.conv.4.weight", "features.7.2.conv.4.bias", "features.7.2.conv.4.running_mean", "features.7.2.conv.4.running_var", "features.7.2.conv.6.weight", "features.7.2.conv.7.weight", "features.7.2.conv.7.bias", "features.7.2.conv.7.running_mean", "features.7.2.conv.7.running_var", "head.weight", "head.bias". 
        
Unexpected key(s) in state_dict: "conv1.weight", "bn1.running_mean", "bn1.running_var", "bn1.weight", "bn1.bias", "layer1.0.conv1.weight", "layer1.0.bn1.running_mean", "layer1.0.bn1.running_var", "layer1.0.bn1.weight", "layer1.0.bn1.bias", "layer1.0.conv2.weight", "layer1.0.bn2.running_mean", "layer1.0.bn2.running_var", "layer1.0.bn2.weight", "layer1.0.bn2.bias", "layer1.0.conv3.weight", "layer1.0.bn3.running_mean", "layer1.0.bn3.running_var", "layer1.0.bn3.weight", "layer1.0.bn3.bias", "layer1.0.downsample.0.weight", "layer1.0.downsample.1.running_mean", "layer1.0.downsample.1.running_var", "layer1.0.downsample.1.weight", "layer1.0.downsample.1.bias", "layer1.1.conv1.weight", "layer1.1.bn1.running_mean", "layer1.1.bn1.running_var", "layer1.1.bn1.weight", "layer1.1.bn1.bias", "layer1.1.conv2.weight", "layer1.1.bn2.running_mean", "layer1.1.bn2.running_var", "layer1.1.bn2.weight", "layer1.1.bn2.bias", "layer1.1.conv3.weight", "layer1.1.bn3.running_mean", "layer1.1.bn3.running_var", "layer1.1.bn3.weight", "layer1.1.bn3.bias", "layer1.2.conv1.weight", "layer1.2.bn1.running_mean", "layer1.2.bn1.running_var", "layer1.2.bn1.weight", "layer1.2.bn1.bias", "layer1.2.conv2.weight", "layer1.2.bn2.running_mean", "layer1.2.bn2.running_var", "layer1.2.bn2.weight", "layer1.2.bn2.bias", "layer1.2.conv3.weight", "layer1.2.bn3.running_mean", "layer1.2.bn3.running_var", "layer1.2.bn3.weight", "layer1.2.bn3.bias", "layer2.0.conv1.weight", "layer2.0.bn1.running_mean", "layer2.0.bn1.running_var", "layer2.0.bn1.weight", "layer2.0.bn1.bias", "layer2.0.conv2.weight", "layer2.0.bn2.running_mean", "layer2.0.bn2.running_var", "layer2.0.bn2.weight", "layer2.0.bn2.bias", "layer2.0.conv3.weight", "layer2.0.bn3.running_mean", "layer2.0.bn3.running_var", "layer2.0.bn3.weight", "layer2.0.bn3.bias", "layer2.0.downsample.0.weight", "layer2.0.downsample.1.running_mean", "layer2.0.downsample.1.running_var", "layer2.0.downsample.1.weight", "layer2.0.downsample.1.bias", "layer2.1.conv1.weight", "layer2.1.bn1.running_mean", "layer2.1.bn1.running_var", "layer2.1.bn1.weight", "layer2.1.bn1.bias", "layer2.1.conv2.weight", "layer2.1.bn2.running_mean", "layer2.1.bn2.running_var", "layer2.1.bn2.weight", "layer2.1.bn2.bias", "layer2.1.conv3.weight", "layer2.1.bn3.running_mean", "layer2.1.bn3.running_var", "layer2.1.bn3.weight", "layer2.1.bn3.bias", "layer2.2.conv1.weight", "layer2.2.bn1.running_mean", "layer2.2.bn1.running_var", "layer2.2.bn1.weight", "layer2.2.bn1.bias", "layer2.2.conv2.weight", "layer2.2.bn2.running_mean", "layer2.2.bn2.running_var", "layer2.2.bn2.weight", "layer2.2.bn2.bias", "layer2.2.conv3.weight", "layer2.2.bn3.running_mean", "layer2.2.bn3.running_var", "layer2.2.bn3.weight", "layer2.2.bn3.bias", "layer2.3.conv1.weight", "layer2.3.bn1.running_mean", "layer2.3.bn1.running_var", "layer2.3.bn1.weight", "layer2.3.bn1.bias", "layer2.3.conv2.weight", "layer2.3.bn2.running_mean", "layer2.3.bn2.running_var", "layer2.3.bn2.weight", "layer2.3.bn2.bias", "layer2.3.conv3.weight", "layer2.3.bn3.running_mean", "layer2.3.bn3.running_var", "layer2.3.bn3.weight", "layer2.3.bn3.bias", "layer3.0.conv1.weight", "layer3.0.bn1.running_mean", "layer3.0.bn1.running_var", "layer3.0.bn1.weight", "layer3.0.bn1.bias", "layer3.0.conv2.weight", "layer3.0.bn2.running_mean", "layer3.0.bn2.running_var", "layer3.0.bn2.weight", "layer3.0.bn2.bias", "layer3.0.conv3.weight", "layer3.0.bn3.running_mean", "layer3.0.bn3.running_var", "layer3.0.bn3.weight", "layer3.0.bn3.bias", "layer3.0.downsample.0.weight", "layer3.0.downsample.1.running_mean", "layer3.0.downsample.1.running_var", "layer3.0.downsample.1.weight", "layer3.0.downsample.1.bias", "layer3.1.conv1.weight", "layer3.1.bn1.running_mean", "layer3.1.bn1.running_var", "layer3.1.bn1.weight", "layer3.1.bn1.bias", "layer3.1.conv2.weight", "layer3.1.bn2.running_mean", "layer3.1.bn2.running_var", "layer3.1.bn2.weight", "layer3.1.bn2.bias", "layer3.1.conv3.weight", "layer3.1.bn3.running_mean", "layer3.1.bn3.running_var", "layer3.1.bn3.weight", "layer3.1.bn3.bias", "layer3.2.conv1.weight", "layer3.2.bn1.running_mean", "layer3.2.bn1.running_var", "layer3.2.bn1.weight", "layer3.2.bn1.bias", "layer3.2.conv2.weight", "layer3.2.bn2.running_mean", "layer3.2.bn2.running_var", "layer3.2.bn2.weight", "layer3.2.bn2.bias", "layer3.2.conv3.weight", "layer3.2.bn3.running_mean", "layer3.2.bn3.running_var", "layer3.2.bn3.weight", "layer3.2.bn3.bias", "layer3.3.conv1.weight", "layer3.3.bn1.running_mean", "layer3.3.bn1.running_var", "layer3.3.bn1.weight", "layer3.3.bn1.bias", "layer3.3.conv2.weight", "layer3.3.bn2.running_mean", "layer3.3.bn2.running_var", "layer3.3.bn2.weight", "layer3.3.bn2.bias", "layer3.3.conv3.weight", "layer3.3.bn3.running_mean", "layer3.3.bn3.running_var", "layer3.3.bn3.weight", "layer3.3.bn3.bias", "layer3.4.conv1.weight", "layer3.4.bn1.running_mean", "layer3.4.bn1.running_var", "layer3.4.bn1.weight", "layer3.4.bn1.bias", "layer3.4.conv2.weight", "layer3.4.bn2.running_mean", "layer3.4.bn2.running_var", "layer3.4.bn2.weight", "layer3.4.bn2.bias", "layer3.4.conv3.weight", "layer3.4.bn3.running_mean", "layer3.4.bn3.running_var", "layer3.4.bn3.weight", "layer3.4.bn3.bias", "layer3.5.conv1.weight", "layer3.5.bn1.running_mean", "layer3.5.bn1.running_var", "layer3.5.bn1.weight", "layer3.5.bn1.bias", "layer3.5.conv2.weight", "layer3.5.bn2.running_mean", "layer3.5.bn2.running_var", "layer3.5.bn2.weight", "layer3.5.bn2.bias", "layer3.5.conv3.weight", "layer3.5.bn3.running_mean", "layer3.5.bn3.running_var", "layer3.5.bn3.weight", "layer3.5.bn3.bias", "layer4.0.conv1.weight", "layer4.0.bn1.running_mean", "layer4.0.bn1.running_var", "layer4.0.bn1.weight", "layer4.0.bn1.bias", "layer4.0.conv2.weight", "layer4.0.bn2.running_mean", "layer4.0.bn2.running_var", "layer4.0.bn2.weight", "layer4.0.bn2.bias", "layer4.0.conv3.weight", "layer4.0.bn3.running_mean", "layer4.0.bn3.running_var", "layer4.0.bn3.weight", "layer4.0.bn3.bias", "layer4.0.downsample.0.weight", "layer4.0.downsample.1.running_mean", "layer4.0.downsample.1.running_var", "layer4.0.downsample.1.weight", "layer4.0.downsample.1.bias", "layer4.1.conv1.weight", "layer4.1.bn1.running_mean", "layer4.1.bn1.running_var", "layer4.1.bn1.weight", "layer4.1.bn1.bias", "layer4.1.conv2.weight", "layer4.1.bn2.running_mean", "layer4.1.bn2.running_var", "layer4.1.bn2.weight", "layer4.1.bn2.bias", "layer4.1.conv3.weight", "layer4.1.bn3.running_mean", "layer4.1.bn3.running_var", "layer4.1.bn3.weight", "layer4.1.bn3.bias", "layer4.2.conv1.weight", "layer4.2.bn1.running_mean", "layer4.2.bn1.running_var", "layer4.2.bn1.weight", "layer4.2.bn1.bias", "layer4.2.conv2.weight", "layer4.2.bn2.running_mean", "layer4.2.bn2.running_var", "layer4.2.bn2.weight", "layer4.2.bn2.bias", "layer4.2.conv3.weight", "layer4.2.bn3.running_mean", "layer4.2.bn3.running_var", "layer4.2.bn3.weight", "layer4.2.bn3.bias", "fc.weight", "fc.bias". 

HTTP Error 403: Forbidden

labels = [-100,2,1]
logits = [[-2.0605, -1.0522,  1.0922],[1.0303, -2.0048,  1.0727],[-1.1031,  1.0414, -2.0464]]
labels_pt = torch.tensor(labels,dtype=torch.int64)
logits_pt = torch.tensor(logits,dtype=torch.float32)

loss = poly_loss(logits_pt,labels_pt,reduction='none')

print('loss',loss)