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:
|
||||
```text
|
||||
Original shape: torch.Size([2, 1000])
|
||||
Pooled shape: torch.Size([2, 1024])
|
||||
```
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user