Efficientnet backbone

H

i noted that only efficientnet-b0 backbone is available for tlt and noted that for pytoch efficient-b4 is available in ngc. Can i used tlt for efficientnet-b4? thanks

Could you please share the link for pytorch efficient-b4 in ngc?

Hi @Morganh
Can you suggest some reference for training efficientnet-b0 (available in NGC) on custom data using tlt?

@imsk
See Overview — Transfer Learning Toolkit 3.0 documentation and Requirements and Installation — Transfer Learning Toolkit 3.0 documentation , you can download faster_rcnn jupyter notebook and try run with efficientnet backbone.

1 Like

hi

i am looking at this:
https://ngc.nvidia.com/catalog/models/nvidia:efficientnet_b4_pyt_amp

That one is not compatible with TLT. In TLT, see Open Model Architectures — Transfer Learning Toolkit 3.0 documentation , then NVIDIA NGC or NVIDIA NGC , currently, only b0 version is available.

thank you. Any chance that other than b0, b4 and above will be made available for transfer learning toolkit?

They are in the roadmap.

1 Like

thank you. could i check on the roadmap what estimated time frame for it? thank you

Yesterday, the b1 versions are available at NVIDIA NGC and
NVIDIA NGC

Currently, EfficientNet B2-B7 are not supported in TLT.

thank you very much. Would hope if other such as B4 can be supported as well since it is already in the ngc.

Can i check what is the differences between those in the TLT and those offered in the NGC(non-TLT)? thanks

In your above-mentioned link NVIDIA NGC , it is for classification. DeepLearningExamples/PyTorch/Classification/ConvNets at master · NVIDIA/DeepLearningExamples · GitHub
It is not compatible with TLT.

In TLT pretrained model link what I mentioned above, they are the pretrained models for TLT classification network or object detection network(FasterRCNN, SSD, DSSD,RetinaNet ) . They are compatible with TLT. Below are the links.
Efficinetnet b0/b1 pretrained models for TLT Classification network: NVIDIA NGC, NVIDIA NGC , NVIDIA NGC,
NVIDIA NGC

Efficinetnet b0/b1 pretrained models for TLT object detection networks: NVIDIA NGC,
NVIDIA NGC,
NVIDIA NGC,
NVIDIA NGC