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

real-test.py #53

Closed dw0127 closed 8 months ago

dw0127 commented 8 months ago

when running the test.py,some situations occured in 'out=model(input,mask_3d_shift,mask_3d_shift_s)',the model that used in test was trainned by me before testting. 微信图片_20240118154502 and the system has been changed into ubuntu,if you could give me some advice?thanks

dw0127 commented 8 months ago

sorry to bother you again,when im trainning in mst_m,'userwarning:the epoch parameter in "scheduler.step()" was not necessary...',is there something wrong?is the trained models count on? Screenshot from 2024-01-19 01-07-36

caiyuanhao1998 commented 8 months ago

(i) You give too much inputs to the trained model.

(ii) No worry

dw0127 commented 8 months ago

(i) You give too much inputs to the trained model.

(ii) No worry

you are so kind!and appreciate your works!!!