Comments (5)
I was able to do some surgery to rework mobilevit classifier to be trainable within Jupyter notebook and customize the fc to match the original ImageNet setup.
It is doing extremely well fine tuning with lr=9e-4 (using Ranger21 optimizer).
If it helps others, here's how I modified the fc (my current work is binary classification so it's 2 instead of 1000).
I think it would be nice to add in this as a simple function call much like how it's done in Timm. I can do a PR for it if interested.
from ml-cvnets.
do you think you could share the jupyter notebook for training?
from ml-cvnets.
https://colab.research.google.com/drive/1xj9SKPL0rntUCNoXHi6GHOOdhaLwfTu2?usp=sharing
from ml-cvnets.
@lessw2020 Inspired by this issue, we have added the functionality of fine-tuning pre-trained classification backbones on other classification datasets in v0.2 of this library. See fine-tuning example here.
from ml-cvnets.
Closing issue because of no activity
from ml-cvnets.
Related Issues (20)
- Questions about the file bytes length of ByteFormer HOT 3
- How to solve this problem:ModuleNotFoundError: No module named 'cvnets.models.classification.' HOT 5
- Normalization Params
- 'nan' loss when training 'ByteFormer' using ImageNet
- Runtime error on single GPU Linux environment training HOT 1
- Not possible to test ByteFormer HOT 5
- Cross Attention Computation in LinearSelfAttention()
- [Feature Request] Docker container
- Segmentation model conversion size mismatch HOT 1
- Size mismatch error when loading a pretrained model HOT 1
- The license of pretrained weights
- Question: Do you have removed the support for video classification? HOT 2
- crash if different number of classess within `train/test` set HOT 6
- VIT-tiny weights and config dont match? HOT 1
- How to convert segmentation model results into VNInstanceMaskObservation? HOT 4
- ModuleNotFoundError: No module named 'main_train' HOT 2
- Using vision transformers for different image resolutions HOT 1
- ModuleNotFoundError: No module named 'main_train'
- ModuleNotFoundError: No module named 'main_train'
- AttributeError: 'NoneType' object has no attribute 'size' HOT 2
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from ml-cvnets.