From be4e0d8f76482087b00e79603b361a9d9728c7c2 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 19 Aug 2023 23:39:09 -0700 Subject: [PATCH] Update attrib comment to include v2 --- timm/models/ghostnet.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/timm/models/ghostnet.py b/timm/models/ghostnet.py index 683595a4..b7c0f5dd 100644 --- a/timm/models/ghostnet.py +++ b/timm/models/ghostnet.py @@ -1,8 +1,11 @@ """ -An implementation of GhostNet Model as defined in: +An implementation of GhostNet & GhostNetV2 Models as defined in: GhostNet: More Features from Cheap Operations. https://arxiv.org/abs/1911.11907 -The train script of the model is similar to that of MobileNetV3 +GhostNetV2: Enhance Cheap Operation with Long-Range Attention. https://proceedings.neurips.cc/paper_files/paper/2022/file/40b60852a4abdaa696b5a1a78da34635-Paper-Conference.pdf + +The train script & code of models at: Original model: https://github.com/huawei-noah/CV-backbones/tree/master/ghostnet_pytorch +Original model: https://github.com/huawei-noah/Efficient-AI-Backbones/blob/master/ghostnetv2_pytorch/model/ghostnetv2_torch.py """ import math from functools import partial