EfficientNet-PyTorch

EfficientNet-PyTorch copied to clipboard

### train is ok ```bash python main.py data/hotDog -b 128 --image_size 300 -a efficientnet-b0 --epochs 300 --gpu 0 ``` ### predict error ```python import torch from torchvision import transforms from...

un-human

Hi, thanks for sharing the code I was wondering what should I change if I want to use a data set with data that can be converted into images of...

Oussamab21

How can the EfficientNet model use in multi-labels image classification?

5

How can the EfficientNet model use in multi-labels image classification?

ljk1072911239

what does line 'pip install -e .' do? I google it ,but there is no answer. Is there anybody konwing? Thank you very much

shaoshuaidu

AttributeError: 'IncompatibleKeys' object has no attribute 'load_state_dict'

1

Hi I am trying to load a model I just trained to continue training from the last step, however I get the following error: AttributeError: 'IncompatibleKeys' object has no attribute...

eborobert

Error when I load the onnx model to the tensorrt format

2

Hi, Thanks for the authors' nice work ! However, when I used the pre-defined efficientnet-b3 model, I have managed to export the onnx model but encountered the intractable problem in...

ShunLu91

Hello. I have a question about function 'extract_endpoints' https://github.com/lukemelas/EfficientNet-PyTorch/blob/master/efficientnet_pytorch/model.py#L231 what is reduction level ? (https://github.com/lukemelas/EfficientNet-PyTorch/blob/master/efficientnet_pytorch/model.py#L240) and I want more explanation about this function. Thanks.

PeterKim1

Efficient net is sensitve to interpolation

1

First, thanks to the author's work, Efficient Net is very useful in my work. And I stumbled upon a problem that this net is very sensitive to interpolation. I think...

NingNanXin

Missing keys when loading pretrained weights

7

`Python 3.8.7` `efficientnet-pytorch==0.7.1` There's an issue loading the pre-trained weights (or any saved weights) when using `include_top = False` ``` from efficientnet_pytorch import EfficientNet model = EfficientNet.from_pretrained('efficientnet-b0', include_top = False)...

BavarianToolbox

KeyError: 'efficientnet-b8'

2

Hello! When I tried: ``` model_name = 'efficientnet-b8' print(EfficientNet.get_image_size(model_name)) Output: 672 ``` Then ``` model = EfficientNet.from_pretrained(model_name) ``` I got error: Traceback (most recent call last): File "myfile.py", line 526,...

ViktorBystrov72