jprdonnelly/MachineLearningNotebooks
1
0
Fork 0
mirror of https://github.com/Azure/MachineLearningNotebooks.git synced 2025-12-19 17:17:04 -05:00
Code Issues Projects Releases Wiki Activity

update samples from Release-129 as a part of SDK release

Browse Source
This commit is contained in:
amlrelsa-ms
2022-03-29 18:28:35 +00:00
parent ceaf82acc6
commit 08b0ba7854
537 changed files with 27112 additions and 151756 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
how-to-use-azureml/reinforcement-learning/cartpole-on-single-compute/files/utils/callbacks.py
View File

@@ -1,23 +0,0 @@
'''RLlib callbacks module:
Common callback methods to be passed to RLlib trainer.
'''
from azureml.core import Run
def on_train_result(info):
'''Callback on train result to record metrics returned by trainer.
'''
run = Run.get_context().parent
run.log(
name='episode_reward_mean',
value=info["result"]["episode_reward_mean"])
run.log(
name='episodes_total',
value=info["result"]["episodes_total"])
run.log(
name='perf_cpu_percent',
value=info["result"]["perf"]["cpu_util_percent"])
run.log(
name='perf_memory_percent',
value=info["result"]["perf"]["ram_util_percent"])