mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
Add missing output
This commit is contained in:
parent
e4360e6125
commit
2b2196028f
@ -90,6 +90,7 @@ print(f'Pooled shape: {o.shape}')
|
|||||||
```
|
```
|
||||||
Output:
|
Output:
|
||||||
```text
|
```text
|
||||||
|
Original shape: torch.Size([2, 1000])
|
||||||
Pooled shape: torch.Size([2, 1024])
|
Pooled shape: torch.Size([2, 1024])
|
||||||
```
|
```
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user