Hi @xchoo I’ve tried different input shapes for EfficientNetB0 and I keep getting errors. I also tried using the default values for the model and that didn’t work either. I did run what you had with DenseNet and it worked, so I tried something similar with EfficientNet and still got errors. Would you happen to have a working model using EfficientNet? Or know where I can find an example that uses it?