DSFD.pytorch icon indicating copy to clipboard operation
DSFD.pytorch copied to clipboard

Is there any results done by using Resnet? Many thanks !

Open math-yyj opened this issue 6 years ago • 6 comments

Many thanks !

math-yyj avatar Jan 16 '19 03:01 math-yyj

@mathyyj have you meet this problem when use the pretrained model. thanks RuntimeError: unexpected EOF, expected 167945 more bytes. The file might be corrupted.

chenjun2hao avatar Jan 16 '19 03:01 chenjun2hao

I am training by using resnet ,but the result is not good ,if the result become better,I will upload resnet model

yxlijun avatar Jan 16 '19 06:01 yxlijun

Hello, is there any result of training by resnet? I am trying to train with a pretrained Resnet152 model import from torchvision, but the weights are not matched. (Since the resnet152 has a final fc layer with 1000 classes), any solution for this? Thanks.

YIYIZH avatar May 15 '19 03:05 YIYIZH

Hello, is there any result of training by resnet? I am trying to train with a pretrained Resnet152 model import from torchvision, but the weights are not matched. (Since the resnet152 has a final fc layer with 1000 classes), any solution for this? Thanks.

You can try to modify the code here in the train.py file as follows:

        if args.model == 'vgg':
            net.vgg.load_state_dict(base_weights)
        else:
            pretrained_dict = base_weights
            model_dict = net.resnet.state_dict()
            pretrained_dict = {k:v for k,v in pretrained_dict.items() if k in model_dict}
            model_dict.update(pretrained_dict)
            print(model_dict)
            net.resnet.load_state_dict(model_dict)

ChauncyFr avatar Jul 31 '19 10:07 ChauncyFr

Hello, is there any result of training by resnet? I am trying to train with a pretrained Resnet152 model import from torchvision, but the weights are not matched. (Since the resnet152 has a final fc layer with 1000 classes), any solution for this? Thanks.

hello,have you fixed the problem and do you get the result by training res152?

pacchomp avatar Jul 03 '23 13:07 pacchomp

这是来自QQ邮箱的假期自动回复邮件。   您好,我最近正在休假中,无法亲自回复您的邮件。我将在假期结束后,尽快给您回复。

chenjun2hao avatar Jul 03 '23 13:07 chenjun2hao