search

caiyuanhao1998 / MST

A toolbox for spectral compressive imaging reconstruction including MST (CVPR 2022), CST (ECCV 2022), DAUHST (NeurIPS 2022), BiSCI (NeurIPS 2023), HDNet (CVPR 2022), MST++ (CVPRW 2022), etc.
MIT License
637 stars 71 forks source link

一轮训练时间 #64

Closed chenxu2002 closed 4 months ago

chenxu2002 commented 4 months ago

用的colab L4GPU，一轮训练大概需要25分钟，请问正常吗？

image
caiyuanhao1998 commented 4 months ago

你好，你训练的是哪个 model 呢？

chenxu2002 commented 4 months ago

你好，你训练的是哪个 model 呢？

你好，你训练的是哪个 model 呢？

您好，我训练的是GAP_Net

caiyuanhao1998 commented 4 months ago

几个stage呢？spatial size 多大？

chenxu2002 commented 4 months ago

用的是您提供的源代码里的模型，9个stage

caiyuanhao1998 commented 4 months ago

9stage 算正常的，别担心

如果觉得我们的repo有用的话，帮忙folk支持一下，感谢