Cadene / Cadene/pretrained-models.pytorch
inceptionv4.py: adaptiveAvgPoolWidth should be an int and not a tensor
Open
- Dominant language
- Python
- Stars
- 9.1k
- Forks
- 1.8k
- PR merge metrics
- No merged PRs in 30d
Description
adaptiveAvgPoolWidth should be an int and not a tensor
put .item() in line 302 of pretrainedmodels/models/inceptionv4.py the following
```
adaptiveAvgPoolWidth = features.shape[2].item()
Contributor guide
No contributing guide indexed for this repository
Research direction
Open pretrainedmodels/models/inceptionv4.py and inspect line 302, where adaptiveAvgPoolWidth is assigned from features.shape[2]. Confirm the value is converted from a tensor to an integer, then verify that InceptionV4 still runs successfully with the updated assignment.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- machine-learning
- Issue type
- Bug
- Difficulty
- 1/5
- Estimated time
- Under an hour
- Activity status
- Stale
- Clarity
- Clearly specified
- Newbie friendliness
- 58/100