Ruo-Ping Dong
f2a8c421
add torch saver test
4 年前
Ruo-Ping Dong
07e82899
update torch saver test
4 年前
GitHub
6f534366
Add torch_utils class, auto-detect CUDA availability ( #4403 )
* Add torch_utils
* Use torch from torch_utils
* Add torch to banned modules in CI
* Better import error handling
* Fix flake8 errors
* Address comments
* Move networks to GPU if enabled
* Switch to torch_utils
* More flake8 problems
* Move reward providers to GPU/CPU
* Remove anothere set default tensor
* Fix banned import in test
4 年前
Ervin Teng
7754ad7b
Don't run value during inference
4 年前
GitHub
b853e5ba
Action buffer ( #4612 )
Co-authored-by: Ervin T <ervin@unity3d.com>
Co-authored-by: Vincent-Pierre BERGES <vincentpierre@unity3d.com>
4 年前
GitHub
8175d558
[bug-fix] Fix BC module + action clipping ( #4667 )
4 年前
vincentpierre
93ca1409
fixing the tests
4 年前
vincentpierre
c1587bce
Solving merge conflicts
4 年前
vincentpierre
1a1070b1
forgot a file
4 年前
vincentpierre
c3699de8
merging master and addressing comments
4 年前
GitHub
d7f549f9
Run pytest on GPU ( #4865 )
* make tests device-friendly
* mark all tests in test_simple_rl
4 年前
Andrew Cohen
6828713c
fix saver test
4 年前
GitHub
f16ce486
Update v2-staging from main (March 15) ( #5123 )
4 年前
GitHub
fc5d0a3f
[bug-fix] Fix save/restore critic, add test ( #5062 )
* Fix save/restore critic, add test
* Rename module for PPO
* Use correct policy in test
4 年前
Ervin Teng
bed4bf36
Load individual elements if state dict load fails ( #5213 )
Co-authored-by: Vincent-Pierre BERGES <vincentpierre@unity3d.com>
Co-authored-by: Ervin T. <ervin@unity3d.com>
(cherry picked from commit ac4f43cf18b98d0fc7063b9b831e07429f7ea39e)
4 年前
Andrew Cohen
98dcb548
test convolutions can be loaded properly
4 年前
Andrew Cohen
2e5b1352
add check that layers still have different dimensions
4 年前
GitHub
cb1f5462
Update ml-agents/mlagents/trainers/tests/torch/saver/test_saver.py
Co-authored-by: Vincent-Pierre BERGES <vincentpierre@unity3d.com>
4 年前