search

dingfengshi / TriDet

[CVPR2023] Code for the paper, TriDet: Temporal Action Detection with Relative Boundary Modeling
MIT License
160 stars 13 forks source link

use_deterministic_algorithms() got an unexpected keyword argument 'warn_only' #2

Closed WalterWangRevo closed 1 year ago

WalterWangRevo commented 1 year ago

Hello, I found line 32 in train_utils.py :use_deterministic_algorithms() got an unexpected keyword argument 'warn_only', my Pytorch version is 1.9.0。

dingfengshi commented 1 year ago

You can simply remove this argument and change this line to torch.use_deterministic_algorithms(True). This argument appeared in pytorch version 1.11. Thank you for your reminder!

dingfengshi commented 1 year ago

Our code needs to run on the version of pytorch1.11. There may be some problems in lower versions.