/ARD

Primary LanguagePython

ARD

This is the code necessary to run the ARD.

Example to run:

python main.py --log_dir=default --device=cuda --dataset=amazon_music

  • --log_dir: file name of the log file.
  • --device: choose cuda to train with gpu, or cpu to train with cpu.

Thanks for using our code!