• xiang song(charlie.song)'s avatar
    [Feature][KG] Multi-GPU training support for DGL KGE (#1178) · bb6a6476
    xiang song(charlie.song) authored
    * multi-gpu
    
    * Pytorch can run but test has acc problem
    
    * pytorch train/eval can run in multi-gpu
    
    * Fix eval
    
    * Fix
    
    * Fix mxnet
    
    * trigger
    
    * triger
    
    * Fix mxnet score_func
    
    * Fix
    
    * check
    
    * FIx default arg
    
    * Fix train_mxnet mix_cpu_gpu
    
    * Make relation mix_cpu_gpu
    
    * delete some dead code
    
    * some opt for update
    
    * Fix cpu grad update
    bb6a6476
train.py 19.1 KB