YtongXie / CoTr

[MICCAI2021] CoTr: Efficiently Bridging CNN and Transformer for 3D Medical Image Segmentation
GNU General Public License v3.0
286 stars 45 forks source link

Directly build and call network #4

Open woodyyyyyyyy opened 1 year ago

woodyyyyyyyy commented 1 year ago

In the process of using other data sets, we want to find the network structure directly. It is found that the network is directly defined on the GPU, which may affect the network modification test or the performance under the condition of multiple GPUs.