如何解决如何正确使用 Tensorboard 进行自定义模型
我创建了非常简单的模型,我正在尝试使用 tensorflow 分析器和 tensorboard 在 GPU 上对其进行分析
import tensorflow as tf
class Layer(tf.keras.layers.Layer):
def __init__(self,dim=8):
super(Layer,self).__init__()
self.dense = tf.keras.layers.Dense(dim)
def call(self,x):
x = self.dense(x)
return x
class CustomModel(tf.keras.Model):
def __init__(self,layer0_dim,layer1_dim):
super(CustomModel,self).__init__()
self.layer0 = Layer(layer0_dim)
self.layer1 = Layer(layer1_dim)
def call(self,x):
y = self.layer0(x)
z = self.layer1(y)
return z
def train():
#tb_callback = tf.keras.callbacks.TensorBoard(log_dir="./logs")
model = CustomModel(8,8)
@tf.function
def step(x):
model(x)
with tf.device("/GPU:0"):
x = tf.random.uniform((4,4))
#tf.profiler.experimental.start('./logs')
with tf.profiler.experimental.Profile('./logs'):
step(x)
#tf.profiler.experimental.stop()
if __name__ == '__main__':
train()
不幸的是,我在张量板上没有看到任何东西。我也没有看到正在生成 tfevents 文件。我在这里做错了什么?
日志目录内容是
logs:
40 Jan 8 15:07 events.out.tfevents.1610147276.test.profile-empty
4096 Jan 8 15:07 plugins
logs/plugins:
4096 Jan 8 15:07 profile
logs/plugins/profile:
4096 Jan 8 15:07 2021_01_08_15_07_56
logs/plugins/profile/2021_01_08_15_07_56:
2023 Jan 8 15:07 test.input_pipeline.pb
4319 Jan 8 15:07 test.kernel_stats.pb
1297 Jan 8 15:07 test.memory_profile.json.gz
3678 Jan 8 15:07 test.overview_page.pb
3316 Jan 8 15:07 test.tensorflow_stats.pb
6478 Jan 8 15:07 test.trace.json.gz
26713 Jan 8 15:07 test.xplane.pb
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。