A fork of https://github.com/lukemelas/EfficientNet-PyTorch with the batch size variable (the bs variable in line 191 of model.py) hard-coded to 1, to make the exported ONNX model compatible with onnx.js.
| Name | Name | Last commit date | ||
|---|---|---|---|---|
A fork of https://github.com/lukemelas/EfficientNet-PyTorch with the batch size variable (the bs variable in line 191 of model.py) hard-coded to 1, to make the exported ONNX model compatible with onnx.js.