search

facebookarchive / fb.resnet.torch

Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts
Other
2.29k stars 664 forks source link

Port to PyTorch #162

Closed panovr closed 7 years ago

panovr commented 7 years ago

Any plan to port this excellent ResNet implementation to PyTorch?

soumith commented 7 years ago

@panovr https://github.com/pytorch/examples/tree/master/imagenet should work fine. We have model definitions and pre-trained models in: https://github.com/pytorch/vision

panovr commented 7 years ago

@soumith Great. May I ask does this imagenet example support model re-train and fine-tune? Because this is current lua/torch resnet repo supported and it is very useful for custom data set.