ml-agents

5 提交

337 分支

128 Plastic标签

目录树: ed33d74a

作者	SHA1	备注	提交日期
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 年前
GitHub	a690af74	[refactor] Make PyTorch the default and TensorFlow optional (#4517 ) * Torch setup.py * Set torch to default * Make torch default in setup.py * Remove indents * Remove other instances of TF being used * Add tensorboard to setup.py * Adding correst setup commands for verifying torch is installed (#4524) * Adding correst setup commands for verifying torch is installed * Editing the test_requirments to add tf and remove torch * Develop torchdefault raise outside setup (#4530) * Torch not imported error to raise at first usage * Torch not imported error to raise at first usage * [refactor] Use PyTorch TensorBoard utils (#4518) * Convert stats writer to use PyTorch TB support * Use common function to print params * Update test * Bump tensorboard to 1.15 to fix the tests * putting tensorboard 1.15.0 as min version requirement Co-authored-by: vincentpierre <vincentpierre@unity3d.com> * [Docs] Initial documentation changes for making...	4 年前
GitHub	457ed0b8	Set torch device from commandline (#4888 )	4 年前

作者

SHA1

备注

提交日期

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 年前

GitHub

a690af74

[refactor] Make PyTorch the default and TensorFlow optional (#4517 )

* Torch setup.py

* Set torch to default

* Make torch default in setup.py

* Remove indents

* Remove other instances of TF being used

* Add tensorboard to setup.py

* Adding correst setup commands for verifying torch is installed (#4524)

* Adding correst setup commands for verifying torch is installed

* Editing the test_requirments to add tf and remove torch

* Develop torchdefault raise outside setup (#4530)

* Torch not imported error to raise at first usage

* Torch not imported error to raise at first usage

* [refactor] Use PyTorch TensorBoard utils (#4518)

* Convert stats writer to use PyTorch TB support

* Use common function to print params

* Update test

* Bump tensorboard to 1.15 to fix the tests

* putting tensorboard 1.15.0 as min version requirement

Co-authored-by: vincentpierre <vincentpierre@unity3d.com>

* [Docs] Initial documentation changes for making...

4 年前

GitHub

457ed0b8

Set torch device from commandline (#4888 )

4 年前

3 次代码提交 (ed33d74a-6b2e-45a7-8047-e75fe85a7f5f)