
EfficientNet-PyTorch:基于PyTorch的EfficientNet实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
EfficientNet-PyTorch是使用PyTorch框架开发的EfficientNet模型的高效实现,适用于图像分类任务。它通过自动模型缩放策略优化了网络结构和参数大小。
使用EfficientNet PyTorch可以通过pip install efficientnet_pytorch命令安装,并通过以下代码加载预训练的模型:
```python
from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_pretrained(efficientnet-b0)
```
更新记录如下:
- 2020年8月25日:新增了一个`include_top(默认为True)`选项,同时提高了代码质量和修复了相关问题。
- 2020年5月14日:增加了全面的注释和文档支持(感谢@workingcoder贡献)。
- 2020年1月23日:基于对抗训练添加了新的预训练模型类别,名为advprop。
全部评论 (0)
还没有任何评论哟~


