/current_want_to_search

Primary LanguagePython

Dependencies

  • torch
  • torch_scatter

Usage

Prepare data.

unzip data.zip

Run training and hyper-parameter searching.

python3 train.py --data_path ./data/WN18RR