[GitHub] [incubator-mxnet] zheng-da opened a new issue #16603: Significant slowdown in some DGL models

[GitBox <gi...@apache.org>]

zheng-da opened a new issue #16603: Significant slowdown in some DGL models
URL: https://github.com/apache/incubator-mxnet/issues/16603
 
 
   I recently compare the performance of DGL KGE models on MXNet 1.5 and the current master branch. I noticed significant slowdown. On MXNet 1.5, it takes 12 seconds to run 1000 batches, and now takes 20 seconds. It seems there is no slowdown on operators after some profiling.
   
   To reproduce the problem, please install DGL 0.4, download the DGL KGE package by cloning [the DGL repo](https://github.com/dmlc/dgl). The DGL KGE package is under apps/kg. Run the following command:
   ```bash
   DGLBACKEND=mxnet python3 train.py --model DistMult --dataset FB15k --batch_size 1024 \
       --neg_sample_size 256 --hidden_dim 2000 --gamma 500.0 --lr 0.1 --max_step 100000 \
       --batch_size_eval 16 --gpu 0 --valid --test -adv
   ```

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services