Skip to content
This repository has been archived by the owner on Jun 13, 2024. It is now read-only.

Add support for caffe-style ResNet and more flexible pretrained model loading #51

Merged
merged 4 commits into from
Nov 12, 2018

Conversation

bowenc0221
Copy link
Contributor

  • Add a config.MODEL.STYLE field in the config to support caffe-style ResNet (by setting config.MODEL.STYLE='caffe')
  • Support more flexible pretrained model loading
    • now support directly loading models trained with PyTorch ImageNet example
    • pretrained model file can be a dictionary with the key: state_dict
    • pretrained model can be directly saved by torch.nn.DataParallel (keys begined with module.)

@msftclas
Copy link

msftclas commented Nov 8, 2018

CLA assistant check
All CLA requirements met.

@bowenc0221
Copy link
Contributor Author

@leoxiaobin

  • add caffe-style ResNet results in README
  • pretrain models are uploaded in Google Drive, you can download these models and put in your shared folder

@leoxiaobin leoxiaobin merged commit 2d723e3 into microsoft:master Nov 12, 2018
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants