dangxusheng
dangxusheng
我认为这个操作和paper里不符合,应该是如下: batch_size, ch = 5, 3 feat_a = torch.randn(batch_size, ch) feat_a.unsqueeze_(-1) feat_b = torch.randn(batch_size, ch) feat_b.unsqueeze_(-1) feat_tmp = torch.cat([feat_a, feat_b], dim=-1) feat_softmax = torch.softmax(feat_tmp, dim=-1) weight_a = feat_softmax[:, :, 0].squeeze()...
root@ys:/home/ys/test# python3.7 test_rknn_lite.py Linux-aarch64 rockchip,rk3568-evb1-ddr4-v10rockchip,rk3568 host: RK356x ************************* None devices connected. ************************* ([], []) ************************************************** Target platforms filled in RKNN model: ['RK3568'] Target platforms supported by this RKNN model: ['RK3566',...