load_model is not None: saved_ckpt_path = os.path.join(os.getcwd(), 'save_model', str(args.load_model)) ckpt = torch.load(saved_ckpt_path) actor.load_state_dict(ckpt['actor']) critic.load_state_dict(ckpt['critic']) discrim.load_state_dict(ckpt['discrim']) running_state.rs.n ...
first_workload_read = False return etcd_snapshot_response eventlet.sleep(10) return None etcd_client.read.side_effect = etcd_client_read with mock.patch.object(agent, 'call_driver') as call_driver: # Notify an endpoint. agent.etcd.on_endpoint_set(EtcdResponse(value=json.dumps({ 'state':...