Skip to content

Commit b0a1976

Browse files
committed
do not use torch.compile for now, still seems to clash with torch_scatter
1 parent 3c19e98 commit b0a1976

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

configs/eval.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ task_name: "eval"
2323
tags: ["dev"]
2424

2525
# compile model for faster training with pytorch >=2.1.0
26-
compile: True
26+
compile: False
2727

2828
# passing checkpoint path is necessary for evaluation
2929
ckpt_path: ???

configs/train.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ train: True
4848
test: True
4949

5050
# compile model for faster training with pytorch >=2.1.0
51-
compile: True
51+
compile: False
5252

5353
# simply provide checkpoint path to resume training
5454
ckpt_path: null

0 commit comments

Comments
 (0)