Skip to content

Commit

Permalink
add default
Browse files Browse the repository at this point in the history
  • Loading branch information
Yiwen Yuan committed Nov 6, 2024
1 parent b43a444 commit 5bc93a2
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions benchmark/tgt_ijcai_benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,6 +57,7 @@
parser.add_argument("--eval_k", type=int, default=10)
parser.add_argument("--gamma_rate", type=int, default=0.95)
parser.add_argument("--seed", type=int, default=42)
parser.add_argument("--result_path", default="results")
args = parser.parse_args()

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
Expand Down

0 comments on commit 5bc93a2

Please sign in to comment.