ml-agents

目录树: 3d2171c4

作者	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	71d1ba68	Set num threads to 1 in Torch (#4436 )	4 年前
Ervin Teng	31675a11	Adjust KMP and interop thread settings	4 年前
Ervin Teng	8cc75388	Changes for experiment	4 年前
Ervin Teng	8fa04c86	Changes for experiment	4 年前
GitHub	da3b0090	[bug-fix] Adjust MKL and KMP Settings (#4443 ) * Adjust KMP and interop thread settings * Don't set affinity, seems to slow down on low-CPU machines	4 年前
GitHub	075b2557	[bug-fix] Set number of threads based on allocated CPU count in Docker containers (#4471 ) * Set num threads properly for Docker * Pylint-friendly logic * Use f.read().rstrip() * Change function names	4 年前
vincentpierre	a899ecff	_	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	7387a77f	remove pylint (#4836 ) * remove pylint * remove other pylint disables	4 年前
GitHub	457ed0b8	Set torch device from commandline (#4888 )	4 年前
GitHub	2e19759c	Turning some logger.info into logger.debug and remove some logging overhead when not using debug (#5211 ) * turning some logger.info into logger.debug and remove some logging overhead when not using debug * Addressing comments * Adding to changelog	4 年前

12 次代码提交 (3d2171c4-e257-4e30-9458-849df69b7176)