I run the source code and get the unexpected result, can somebody help ?

[yulong-CSAI]

I start to read the dglke source code, and decide to develep on this code, However, when I run the source code , I got the unexpected result.
I run the source code by this command:
python train.py --model_name TransE_l2 --dataset FB15k --batch_size 1000 --neg_sample_size 200 --hidden_dim 400 --gamma 19.9 --lr 0.25 --max_step 500 --log_interval 100 --batch_size_eval 16 --test -adv --regularization_coef 1.00E-09 --num_thread 1 --num_proc 48
the result is:

[yulong-CSAI]

can sb help?

[yulong-CSAI]

did I miss something?

[classicsong]

How many cpus do you have?

[classicsong]

If you have 24 cores, you should use

dglke_train --model_name TransE_l2 --dataset FB15k --batch_size 1000 --neg_sample_size 200 --hidden_dim 400 --gamma 19.9 --lr 0.25 --max_step 500 --log_interval 100 --batch_size_eval 16 --test -adv --regularization_coef 1.00E-09 --num_thread 1 --num_proc 24

You need to change num_proc and the max_step.

[yulong-CSAI]

If you have 24 cores, you should use

dglke_train --model_name TransE_l2 --dataset FB15k --batch_size 1000 --neg_sample_size 200 --hidden_dim 400 --gamma 19.9 --lr 0.25 --max_step 500 --log_interval 100 --batch_size_eval 16 --test -adv --regularization_coef 1.00E-09 --num_thread 1 --num_proc 24

You need to change num_proc and the max_step.

Thanks to reply，It works !!