Fix the saving issue for HAN

karkaroff / castor

PyTorch deep learning models by the Data Systems Group at the University of Waterloo

http://castor.ai/

Apache License 2.0

0 stars 0 forks source link

Closed Ashutosh-Adhikari closed 5 years ago

Ashutosh-Adhikari commented 5 years ago

Hi @achyudhk , this fix shall allow us to use torch.save(model....) consistently even with HAN. Please approve.

achyudh commented 5 years ago

LGTM.