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-170 as a part of 1.51 SDK stable release

Browse Source
This commit is contained in:
amlrelsa-sa
2023-06-06 10:50:33 +05:30
parent 579643326d
commit 3a1824e3ad
46 changed files with 1669 additions and 1382 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,21 +3,20 @@
'''
from azureml.core import Run
from ray import tune
from ray.tune import Callback
from ray.air import session
def on_train_result(info):
'''Callback on train result to record metrics returned by trainer.
'''
run = Run.get_context()
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"])
class TrialCallback(Callback):
def on_trial_result(self, iteration, trials, trial, result, **info):
'''Callback on train result to record metrics returned by trainer.
'''
run = Run.get_context()
run.log(
name='episode_reward_mean',
value=result["episode_reward_mean"])
run.log(
name='episodes_total',
value=result["episodes_total"])