From 9cc119e129b896779f98cccab48d9f944911b5c6 Mon Sep 17 00:00:00 2001 From: dvanaken Date: Fri, 17 Apr 2020 00:07:04 -0400 Subject: [PATCH] Added flag to alt_create_or_edit_session() that specifies whether to return the ddpg models --- server/website/website/views.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/server/website/website/views.py b/server/website/website/views.py index f7c98f6..ee284bb 100644 --- a/server/website/website/views.py +++ b/server/website/website/views.py @@ -1553,6 +1553,7 @@ def alt_create_or_edit_session(request): session_knobs = data.pop('session_knobs', None) disable_others = data.pop('disable_others', False) hyperparams = data.pop('hyperparameters', None) + return_ddpg_model = data.pop('return_ddpg_model', False) ts = now() if request.path == reverse('backdoor_create_session'): @@ -1634,10 +1635,16 @@ def alt_create_or_edit_session(request): res['hardware_id'] = res['hardware'] res['hardware'] = model_to_dict(session.hardware) res['algorithm'] = AlgorithmType.name(res['algorithm']) - if session.ddpg_actor_model is not None: - res['ddpg_actor_model'] = base64.encodebytes(session.ddpg_actor_model).decode('utf8') - res['ddpg_critic_model'] = base64.encodebytes(session.ddpg_critic_model).decode('utf8') - res['ddpg_replay_memory'] = base64.encodebytes(session.ddpg_reply_memory).decode('utf8') + if return_ddpg_model: + if session.ddpg_actor_model is not None: + res['ddpg_actor_model'] = base64.encodebytes(session.ddpg_actor_model).decode('utf8') + res['ddpg_critic_model'] = base64.encodebytes(session.ddpg_critic_model).decode('utf8') + res['ddpg_replay_memory'] = base64.encodebytes( + session.ddpg_reply_memory).decode('utf8') + else: + res['ddpg_actor_model'] = None + res['ddpg_critic_model'] = None + res['ddpg_replay_memory'] = None sk = SessionKnob.objects.get_knobs_for_session(session) sess_knobs = {} for knob in sk: