未验证 提交 05770197 编写于 作者: W Wei Shengyu 提交者: GitHub

Merge pull request #1855 from flytocc/VAN_Classification

add paper citation
# copyright (c) 2021 PaddlePaddle Authors. All Rights Reserve.
# copyright (c) 2022 PaddlePaddle Authors. All Rights Reserve.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
......@@ -185,6 +185,11 @@ class OverlapPatchEmbed(nn.Layer):
class VAN(nn.Layer):
r""" VAN
A PaddlePaddle impl of : `Visual Attention Network` -
https://arxiv.org/pdf/2202.09741.pdf
"""
def __init__(self,
img_size=224,
in_chans=3,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册