将损失的数值写入SummaryWriter对象是加在训练代码里嘛

是的，将损失的数值写入SummaryWriter对象的代码通常会放在训练代码中。具体来说，你需要在每个训练batch或每个训练epoch结束时，获取当前的损失值，并使用`add_scalar`方法将其写入SummaryWriter对象。

下面是一个示例，展示了如何将每个训练batch的损失值写入SummaryWriter对象：

from tensorboardX import SummaryWriter

# 创建SummaryWriter对象
writer = SummaryWriter()

# 在训练过程中，将每个batch的损失值写入SummaryWriter对象
for batch in data_loader:
    # 前向传播
    output = model(batch)
    
    # 计算损失
    loss = compute_loss(output, batch.target)
    
    # 将损失值写入SummaryWriter对象
    writer.add_scalar('Loss/train_batch', loss.item(), global_step)

# 关闭SummaryWriter对象
writer.close()

在上述示例中，我们使用`add_scalar`方法将每个batch的损失值写入名为`Loss/train_batch`的标量数据。`loss.item()`获取损失值的纯数量值，`global_step`表示当前是训练过程中的第几个batch。

你可以根据需要将这段代码放在你的训练循环中的适当位置，以记录和可视化损失值。同样地，你也可以在每个训练epoch结束时将损失值写入SummaryWriter对象，只需稍作修改即可。

希望这个示例能帮助你在训练过程中将损失的数值写入SummaryWriter对象！