Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- IEnumerator UpdateObs()
- {
- for (var i = 0; i < currentInputs.Length / 3; i++)
- {
- var idx_input = i * 3;
- //var idx_output = i * 2;
- Dictionary<string, Tensor> inputs = new Dictionary<string, Tensor>();
- float[] input_data = new float[]
- {
- currentInputs[ idx_input + 0 ],
- currentInputs[ idx_input + 1 ],
- currentInputs[ idx_input + 2 ]
- };
- var input_1_1 = new Tensor(1, 3, input_data);
- inputs.Add("input_1_1", input_1_1); // "input_1"
- var coroutine = Worker_instance.ExecuteAsync(inputs);
- yield return StartCoroutine(coroutine);
- // alternatevly use wroker.peek( "layer_name" )
- Tensor output = Worker_instance.Fetch(Model_instance.outputs[0]); // "dense_1_1/Relu" , "dense_1"
- currentOutputs = output.data.Download(Model_output_size);
- agent.UpdateInputs(currentOutputs);
- yield return new WaitForSeconds(0.01f);
- foreach (var input in inputs)
- input.Value.Dispose();
- output.Dispose();
- }
- Worker_instance.Dispose();
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement