test_urpc.py about ssl4mis HOT 3 CLOSED

Hi,
Please check this line.
Best,
Xiangde.

from ssl4mis.

Thanks for your reply.

from ssl4mis.

Thanks for your reply.

Hi,
Did you successfully run the 2D 'test_urpc.py' ? I met the problem 'Traceback (most recent call last):
File "/home/b417/xb/5_SSL4MIS-master/SSL4MIS-master/code/test_2D_fully.py", line 121, in
metric = Inference(FLAGS)
File "/home/b417/xb/5_SSL4MIS-master/SSL4MIS-master/code/test_2D_fully.py", line 110, in Inference
case, net, test_save_path, FLAGS)
File "/home/b417/xb/5_SSL4MIS-master/SSL4MIS-master/code/test_2D_fully.py", line 64, in test_single_volume
out_main, dim=1), dim=1).squeeze(0)
TypeError: softmax() received an invalid combination of arguments - got (tuple, dim=int), but expected one of:

• (Tensor input, name dim, *, torch.dtype dtype)
• (Tensor input, int dim, torch.dtype dtype)'. I am wondering if you know how to solve it
  
  
  .

from ssl4mis.