tensorflow - 如何从 tensorflow_hub 查看模型中的修改层

Question

我正在尝试按照 https://www.tensorflow.org/hub/tutorials/tf_hub_generation_image_module上的步骤构建 GAN 。

它用于tensorflow_hub.load加载预训练模型。

progan = hub.load("https://tfhub.dev/google/progan-128/1").signatures['default']

progan 的类型是一个函数。

print(type(progan))
>>> <class 'tensorflow.python.eager.wrap_function.WrappedFunction'>```

for i in progan.__dict__:
  print(i)
>>> _variable_holder
_signature
_arg_keywords
_num_positional_args
_func_graph
_captured_inputs
_captured_closures
_function_spec
_pre_initialized_function_spec
_output_shapes
_attrs
_garbage_collector
_delayed_rewrite_functions
_first_order_tape_functions
_higher_order_tape_functions
_inference_function

如果我想获取模型并添加或删除一些图层，我应该怎么做？

标签： tensorflow