1. 论文模型图:
2. 模型代码:pytorch 实现
https://github.com/oukohou/SSR_Net_Pytorch/blob/master/SSR_models/SSR_Net_model.py
注:pytorch 中的 view 在 onxx 的可视化 Netron 中 Reshape;Linear 为 Gemm;切片 为 Gather。
模型架构图:
细节:
图中的各个模块的细节请参考开源代码中的具体实现。
标签:模型,细节,SSR,pytorch,Net,解析 From: https://www.cnblogs.com/odesey/p/17101233.html