search

OpenGVLab / Vision-RWKV

Vision-RWKV: Efficient and Scalable Visual Perception with RWKV-Like Architectures
https://arxiv.org/abs/2403.02308
Apache License 2.0
371 stars 14 forks source link

关于T_MAX的问题 #38

Open chenzean opened 2 months ago

chenzean commented 2 months ago

作者，你好！ 我想问一下T_MAX 这个设置是否和图像patch有关？如果图像的patch是96是不是需要时96*96？

chenzean commented 2 months ago

image 可以这样导入两种WKV吗？ 导入两种WKV后会显示： Process finished with exit code 134

duanduanduanyuchen commented 2 weeks ago

可以这样导入，不过你需要把wkv_op.cpp里面 TORCH_LIBRARY(wkv, m) 的wkv修改成load函数里面的name