ValueError: Unsupported architecture: efficientnet

hii @nvidians

I am trying to train a efficient net_b0_swish classification model.


I am passing architecture name like this but it says :
ValueError: Unsupported architecture: efficientnet ??

Any help will be appreciable.
Thanks

Which version of TLT did you use?

@Morganh
tlt-streamanalytics:v2.0_py3

TLT 2.0 does not support efficientnet.
TLT 3.0 can support efficientnet.
See Open Model Architectures — Transfer Learning Toolkit 3.0 documentation

thanks a lot @Morganh