use_deterministic_algorithms() got an unexpected keyword argument 'warn_only'
WalterWangRevo opened this issue · 2 comments
WalterWangRevo commented
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
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
Our code needs to run on the version of pytorch1.11. There may be some problems in lower versions.