2679622694

When loading model parameters during training, the program has been stuck without response 

RuntimeError: CUDA error: out of memory

1

This is problem feedback： Traceback (most recent call last): File "main.py", line 188, in main() File "main.py", line 167, in main outputs = model(inputs.unsqueeze(0)) File "/home/zjh/anaconda3/envs/psmnet/lib/python3.7/site-packages/torch/nn/modules/module.py", line 477, in __call__...

How to visualize the result

2

Thank you for your excellent work. How to visualize the results？