使用tensorboard可视化训练

from torch.utils.tensorboard import SummaryWriter
witer=SummaryWriter('logs')
for i in range(100):
    witer.add_scalar("y=x",2*i,i)
witer.close()

然后在终端里

tensorboard --logdir=绝对路径 --port=6007 

就可以进入tensorboard查看

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
下面是一个使用TensorBoard可视化训练过程的示例代码,其中包含了输入图片进行可视化的代码框架: ```python import tensorflow as tf from tensorflow.keras.layers import Input, Conv2D, BatchNormalization, Activation, MaxPooling2D, Flatten, Dense from tensorflow.keras.callbacks import TensorBoard # 定义模型结构 def model(input_shape, num_classes): input_tensor = Input(shape=input_shape) x = Conv2D(64, kernel_size=(3, 3), strides=(1, 1), padding='same')(input_tensor) x = BatchNormalization()(x) x = Activation('relu')(x) x = MaxPooling2D(pool_size=(2, 2))(x) x = Conv2D(128, kernel_size=(3, 3), strides=(1, 1), padding='same')(x) x = BatchNormalization()(x) x = Activation('relu')(x) x = MaxPooling2D(pool_size=(2, 2))(x) x = Flatten()(x) x = Dense(num_classes, activation='softmax')(x) model = tf.keras.models.Model(inputs=input_tensor, outputs=x) return model # 加载数据集 (x_train, y_train), (x_test, y_test) = tf.keras.datasets.mnist.load_data() x_train = x_train.reshape(-1, 28, 28, 1) / 255.0 x_test = x_test.reshape(-1, 28, 28, 1) / 255.0 y_train = tf.keras.utils.to_categorical(y_train, num_classes=10) y_test = tf.keras.utils.to_categorical(y_test, num_classes=10) # 定义模型参数 input_shape = (28, 28, 1) num_classes = 10 batch_size = 128 epochs = 10 # 定义TensorBoard回调函数 tb_callback = TensorBoard(log_dir='./logs', histogram_freq=1, write_graph=True, write_images=True) # 定义模型 model = model(input_shape, num_classes) model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy']) # 训练模型,并使用TensorBoard可视化训练过程 model.fit(x_train, y_train, batch_size=batch_size, epochs=epochs, validation_data=(x_test, y_test), callbacks=[tb_callback]) ``` 该代码定义了一个简单的卷积神经网络模型,使用MNIST数据集进行训练,并使用TensorBoard可视化训练过程。其中,`model`函数定义了模型结构,`TensorBoard`回调函数定义了可视化的参数,例如记录日志的路径、直方图的频率和是否记录模型图等。在训练模型时,将`TensorBoard`回调函数传递给`fit`方法的`callbacks`参数,即可实现使用TensorBoard可视化训练过程。此外,为了输入图片进行可视化,还需要在`TensorBoard`回调函数中将`write_images`参数设置为`True`,这样训练过程中的输入图片会被记录到TensorBoard日志中。最后,在终端中运行以下命令即可启动TensorBoard进行可视化: ``` tensorboard --logdir=./logs ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值