Thank you for your reply.
I tried to save the full model by adding:
torch.save(net,‘mb_v1_ssd_all.pth’)
only to find this make an error:
File “/home/hong/anaconda3/envs/PyTorch1.0_GPU_ONNX/lib/python3.7/site-packages/torch/serialization.py”, line 291, in _save
pickler.dump(obj)
TypeError: can’t pickle module objects
so,it can not be saved in this way?