improve transformer architecture, remove 3.10 install constraint, add documentation for torch.compile()

2023-05-06 16:12:10 +00:00
parent af139ffbab
commit 3bbb7e38ea
3 changed files with 36 additions and 16 deletions
@@ -2,7 +2,7 @@
 -r requirements-freqai.txt

 # Required for freqai-rl
-torch==2.0.0; python_version < '3.11'
+torch==2.0.0
 #until these branches will be released we can use this
 gymnasium==0.28.1
 stable_baselines3==2.0.0a5