I really admire your work on ICGNet. Would you consider to open source the code for the training part?