Skip to content

Commit

Permalink
add comment on compile
Browse files Browse the repository at this point in the history
  • Loading branch information
segasai committed Nov 17, 2024
1 parent 4a2d9fe commit 2a75593
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions py/rvspecfit/nn/train_interpolator.py
Original file line number Diff line number Diff line change
Expand Up @@ -192,6 +192,7 @@ def main(args):
print('loss0', loss0 / spread0)

myint.to(train_dev)
# myint.compile() slows down
tD_0 = tD_0.to(train_dev)
tSD_0 = tSD_0.to(train_dev)

Expand Down

0 comments on commit 2a75593

Please sign in to comment.