zheng-da commented on issue #16603: Significant slowdown in some DGL models URL: https://github.com/apache/incubator-mxnet/issues/16603#issuecomment-549542817 I just tried the experiment again and there is no problem. The command to run the experiment: ``` 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 2000 --gpu 0 ``` You can use the following commands to install MXNet. The problem is very easy to reproduce. You can install the MKLDNN version if you want. It makes no difference. ``` pip3 install mxnet-cu100 ``` ``` pip3 install --pre mxnet-cu100 ```
---------------------------------------------------------------- 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: us...@infra.apache.org With regards, Apache Git Services
