HTensor

same issue. this may be caused by internal implementation in DataParallel. To solve it, you can simply rename ```forward_split_cat``` to ```forward``` and comment some codes in ```__init__```.