mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Summary: add an option to use a resnet network instead of alexnet. Modified the resnet.create_resnet50 function slightly to allow specifying different kernel/stride parameters so we can adapt resnet to our image size. Differential Revision: D4472535 fbshipit-source-id: ed06acf52f6425a1e04d047548eb3c70388d74aa |
||
|---|---|---|
| .. | ||
| resnet.py | ||