The dataset are available in Google Drive
python main_div.py --data music_small --lr 0.001 --wd 1e-5 --KG_layers 5 --n_layers 1 --gamma 0.9 --beta 0.4