未验证 提交 d8fbc1b3 编写于 作者: C cuicheng01 提交者: GitHub

Merge pull request #1407 from cuicheng01/develop

update vision_transformer.py
...@@ -12,6 +12,8 @@ ...@@ -12,6 +12,8 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
# Code was based on https://github.com/rwightman/pytorch-image-models/blob/master/timm/models/vision_transformer.py
from collections.abc import Callable from collections.abc import Callable
import numpy as np import numpy as np
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册