search

nagejacob / TBSN

GNU General Public License v3.0
24 stars 1 forks source link

TypeError when validate #1

Closed ZKCCZ closed 4 months ago

ZKCCZ commented 4 months ago

Thanks for your excerllent works! Here's an error when I run 

cd validate
python base.py --config_file "../option/tbsn_sidd.json"

Error:

TypeError: zeros() received an invalid combination of arguments - got (float, float), but expected one of:
 * (tuple of ints size, *, tuple of names names, torch.dtype dtype, torch.layout layout, torch.device device, bool pin_memory, bool requires_grad)
 * (tuple of ints size, *, Tensor out, torch.dtype dtype, torch.layout layout, torch.device device, bool pin_memory, bool requires_grad)
ZKCCZ commented 4 months ago

it seems that the error appears when initial model with default setting overlap_ratio=[0.5, 0.5, 0.5] Is it right to change it to [0,0,0]?

nagejacob commented 4 months ago

Please upload more complete output rather than the error message, I run this code with no problem in my computer.