tonhathuy
Results
5
comments of
tonhathuy
It's working for me ` model.load_state_dict(torch.load(opt.checkpoint_model,map_location=device),strict=False)`
I follow your tutorial, my code is this: ``` model_dict = model.state_dict() if conf.resume: ori_epoch = torch.load(args.pretrain_model)['epoch'] + 1 pretrained_dict = torch.load(args.pretrain_model)['state_dict'] print(pretrained_dict.keys()) new_pretrained_dict = {} for k in model_dict:...
I converted this model to onnx and tensorRT , checkout [my repo](https://github.com/tonhathuy/tensorrt-triton-magface).
`pip install timm`
@Debugerss I have a same error, have you fixed it yet ?