浏览代码

remove unused variables

/develop-generalizationTraining-TrainerController
Chris Elion 6 年前
当前提交
85809f78
共有 1 个文件被更改,包括 0 次插入4 次删除
  1. 4
      ml-agents/mlagents/trainers/tensorflow_to_barracuda.py

4
ml-agents/mlagents/trainers/tensorflow_to_barracuda.py


auto_pad = get_attr(layer, "padding") # layer.attr['padding'].s.decode("utf-8")
pads = get_attr(layer, "pads")
strides = get_attr(layer, "strides") # layer.attr['strides'].list.i
# TODO remove?
# dilations = get_attr(layer, "dilations") # layer.attr['dilations'].list.i
pool_size = get_attr(layer, "ksize") # layer.attr['ksize'].list.i
shape = get_attr(layer, "shape")
starts = get_attr(layer, "starts")

o_model.layers = cleanup_layers(o_model.layers)
all_inputs = {i for l in o_model.layers for i in l.inputs}
# TODO remove?
# embedded_tensors = {t.name for l in o_model.layers for t in l.tensors}
# Trim
if trim_unused_by_output:

正在加载...
取消
保存