EfficientNet-PyTorch
EfficientNet-PyTorch copied to clipboard
lukemelas
A PyTorch implementation of EfficientNet
I'm trying to script the EfficientNet module with torchscript but there's an error while MemoryEfficientSwish seems to be not supported with torchscript. Have anyone successfully scripted that module yet?
Hello, I want to ask how I can train my own data set now. I don't see the training code
Hello. I am facing this odd problem with EfficientNet. I am trying to pretrain it from scratch using four classes from the `CIFAR10`. But the validation accuracy remains extremely low...
Could you please guide me on how to add a multi-head attention layer after model. features[5:] . I want to add this layer inside the features . import torch import...
The code is found on line 475 of [EfficientNet-PyTorch/efficientnet_pytorch/utils.py](https://github.com/lukemelas/EfficientNet-PyTorch/blob/master/efficientnet_pytorch/utils.py). According to formula 3 of the original paper, the image size of efficientnet-b7 is 600. _r=γ^Φ=1.15^Φ=600/224=2.6786_, so _Φ=log(2.6786)/log(1.15)=7.04_. Therefore, _depth=d=α^Φ=1.2^7.04=3.6094≠3.1_ I...
How to train a new model on the imagenet dataset？
Has anyone trained EfficientNet-B0 from scratch on ImageNet and successfully reproduced the results? I used the model in this repo and tried to follow the hyperparameters as closely as I...
when i convert to onnx,
Hi, I would like to determine whether your training strategy includes NoisyStudent and other data augmentation. Please, reply me, thank you!
is it possible to do multilabel and multiclass image classification with efficientnet b5? i googled around and there's a page saying in order to do multilabeling, we need to change...
