Skip to content

Fix Efficientnet enocoder for torchscript #1037

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Jan 16, 2025
+145 −198
Merged

Fix Efficientnet enocoder for torchscript #1037

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
ce1ae43
Refactor effnet for torchscript
qubvel Jan 16, 2025
4aa7cd1
Update encoder
qubvel Jan 16, 2025
22e6b2e
Update docs
qubvel Jan 16, 2025
ebe45dc
Update indexes
qubvel Jan 16, 2025
1ad6d80
Fix
qubvel Jan 16, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 8 additions & 8 deletions docs/encoders.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -99,14 +99,14 @@ densenet201 imagenet 18M
densenet161 imagenet 26M ✅ ✅ ✅
inceptionresnetv2 imagenet / imagenet+background 54M ✅ ✅ ✅
inceptionv4 imagenet / imagenet+background 41M ✅ ✅ ✅
efficientnet-b0 imagenet / advprop 4M ❌ ✅ ✅
efficientnet-b1 imagenet / advprop 6M ❌ ✅ ✅
efficientnet-b2 imagenet / advprop 7M ❌ ✅ ✅
efficientnet-b3 imagenet / advprop 10M ❌ ✅ ✅
efficientnet-b4 imagenet / advprop 17M ❌ ✅ ✅
efficientnet-b5 imagenet / advprop 28M ❌ ✅ ✅
efficientnet-b6 imagenet / advprop 40M ❌ ✅ ✅
efficientnet-b7 imagenet / advprop 63M ❌ ✅ ✅
efficientnet-b0 imagenet / advprop 4M ✅ ✅ ✅
efficientnet-b1 imagenet / advprop 6M ✅ ✅ ✅
efficientnet-b2 imagenet / advprop 7M ✅ ✅ ✅
efficientnet-b3 imagenet / advprop 10M ✅ ✅ ✅
efficientnet-b4 imagenet / advprop 17M ✅ ✅ ✅
efficientnet-b5 imagenet / advprop 28M ✅ ✅ ✅
efficientnet-b6 imagenet / advprop 40M ✅ ✅ ✅
efficientnet-b7 imagenet / advprop 63M ✅ ✅ ✅
mobilenet_v2 imagenet 2M ✅ ✅ ✅
xception imagenet 20M ✅ ✅ ✅
timm-efficientnet-b0 imagenet / advprop / noisy-student 4M ✅ ✅ ✅
Expand Down
Loading
Loading