mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
c7c09722ad
Context: https://github.com/pytorch/torchdynamo/issues/1588 This PR moves [TorchDynamo](https://github.com/pytorch/torchdynamo) and TorchInductor into PyTorch core. - `torchdynamo` becomes `torch._dynamo` - `torchinductor` becomes `torch._inductor` This PR was generated by running `copy_to_core.sh` in https://github.com/pytorch/torchdynamo/pull/1538 Pull Request resolved: https://github.com/pytorch/pytorch/pull/86461 Approved by: https://github.com/voznesenskym
27 lines
901 B
Python
27 lines
901 B
Python
# resnet50 layer shape
|
|
resnet50_layers = (
|
|
# IN_H, IN_W, IN_C, KERNEL_H, KERNEL_W, KERNEL_N, stride, padding
|
|
(224, 224, 3, 7, 7, 64, (2, 2), (0, 0)),
|
|
# conv2_x
|
|
(56, 56, 64, 1, 1, 64, (1, 1), (0, 0)),
|
|
(56, 56, 64, 3, 3, 64, (1, 1), (0, 0)),
|
|
(56, 56, 64, 1, 1, 256, (1, 1), (0, 0)),
|
|
# conv3_x
|
|
(56, 56, 256, 1, 1, 128, (2, 2), (0, 0)),
|
|
(28, 28, 128, 3, 3, 128, (1, 1), (0, 0)),
|
|
(28, 28, 128, 1, 1, 512, (1, 1), (0, 0)),
|
|
# conv4_x
|
|
(28, 28, 512, 1, 1, 256, (2, 2), (0, 0)),
|
|
(14, 14, 256, 3, 3, 256, (1, 1), (0, 0)),
|
|
(14, 14, 256, 1, 1, 1024, (1, 1), (0, 0)),
|
|
# conv5_x
|
|
(14, 14, 1024, 1, 1, 512, (2, 2), (0, 0)),
|
|
(7, 7, 512, 3, 3, 512, (1, 1), (0, 0)),
|
|
(7, 7, 512, 1, 1, 2048, (1, 1), (0, 0)),
|
|
)
|
|
|
|
alexnet_layers = (
|
|
# IN_H, IN_W, IN_C, KERNEL_H, KERNEL_W, KERNEL_N, stride, padding
|
|
(224, 224, 3, 11, 11, 64, (4, 4), (2, 2)),
|
|
)
|