entityMap|blocks|key|1cjjm|text|当我删除带有回调的行时，summary_writer起作用了，并在google+colab上添加了以下内容：|type|unstyled|depth|inlineStyleRanges|entityRanges|data^0^^$0|$]|1|@$2|3|4|5|6|7|8|C|9|@]|A|@]|B|$]]]]

Worked when I removed the line with callbacks, summary_writer and instead added this on google colab:
<pre><code>%load_ext tensorboard
%cd '/content/drive/MyDrive/path_to_the_notebook_/'
# *train* folder is created by trax and holds the logs for the train run
# to log the eval run change to --logdir eval
%tensorboard --logdir train # train = name of folder
</code></pre>

Anyone managed to log the loss with TensorBoard? I am using the trax ml library.
I am getting this error <code>TypeError: 'SummaryWriter' object is not callable</code>.
I am using the <code>SummaryWriter</code> from <code>jaxboard</code> and then adding it to <code>callbacks</code> within <code>training.Loop</code>.
<pre><code>my_dir = &quot;/some_dir&quot; + datetime.datetime.now().strftime(&quot;%Y%m%d-%H%M%S&quot;)
summary_writer = jaxboard.SummaryWriter(log_dir=my_dir)

def train_model(model, batch_size=batch_size, n_steps=1000, output_dir=output_dir):
 '''
 Input: 
 model - the model
 train_task - Training task
 eval_tasks - Evaluation task
 n_steps - the evaluation steps
 output_dir - folder to save your files
 Output:
 trainer - trax trainer
 '''
 train_task = training.TrainTask(
 labeled_data=train_generator(batch_size=batch_size, shuffle=True),
 loss_layer=TripletLoss(),
 optimizer=trax.optimizers.Adam(learning_rate=0.001),
 n_steps_per_checkpoint=1000,
 )

 eval_tasks = training.EvalTask(
 labeled_data=val_generator(batch_size=batch_size, shuffle=True),
 metrics=[TripletLoss()],
 n_eval_batches=10,
 )

 training_loop = training.Loop(
 model, # The learning model
 train_task, # The training task
 eval_tasks = eval_tasks, # The evaluation task
 #random_seed=35,
 output_dir = output_dir, # The output directory
 callbacks=[summary_writer], # Logging
 ) 
 
 training_loop.run(n_steps = n_steps)
 
 # Return the training_loop, since it has the model.
 return training_loop
</code></pre>
The error appears when I run the training loop:
<pre><code>training_loop = train_model(my_model())
</code></pre>

TensorBoard with Trax

翻译质量差，导致语言生硬或混乱。

没有提供实际的解决方法或示例。

解答不清晰，无法理解或解决问题。

页面排版不美观，阅读体验差。

文章

问答

视频

教程

学习中心

腾讯云实验室

直播

竞赛

腾讯云代码分析专区

腾讯iOA零信任安全管理系统专区

腾讯云架构师技术同盟交流圈

腾讯云数据库专区

腾讯云智能顾问专区

腾讯云原生专区

腾讯混元专区

腾讯云TCE专区

腾讯云Lighthouse专区

腾讯云HAI专区

腾讯云Edgeone专区

腾讯云存储专区

腾讯云智能专区

腾讯轻联专区 

腾讯云开发专区

TAPD专区

腾讯轻量云游戏服专区

腾讯云最具价值专家

腾讯云架构师技术同盟

腾讯云创作之星

腾讯云开发者先锋

腾讯云AI代码助手

云原生构建

TAPD 敏捷项目管理

Cloud Studio

SDK中心

API中心

命令行工具

功能1上新10个字符

功能2描述100个字符功能2描述100个字符功能2描述100个字符功能2描述100个字符功能2描述100个字符功能2描述100个字符功能2描述100个字符功能2描述100个字符功能2描述100个字符。

功能2上新100个字符功能2上新100个字符功能2上新100个字符功能2上新100个字符功能2上新100个字符功能2上新100个字符功能2上新100个字符功能2上新100个字符功能2上新100个字符。

功能5描述100个字符功能5描述100个字符功能5描述100个字符功能5描述100个字符功能5描述100个字符功能5描述100个字符

功能5上新100个字符功能5上新100个字符功能5上新100个字符功能5上新100个字符功能5上新100个字符功能5上新100个字符功能5上新100个字符功能5上新100个字符功能5上新100个字符功能5上新100个字符

功能4上新

文章&问答评论现已支持表情

全新交互，全新视觉，新增快捷键、悬浮工具栏、高亮块等功能并同时优化现有功能，全面提升创作效率和体验

社区富文本编辑器全新改版！诚邀体验～ 

精选全网热门MCP server，让你的AI更好用 🚀

💥开发者 MCP广场重磅上线！

涵盖代码开发、场景应用、自动测试全流程，助你从零构建专属AI助手

一站式MCP教程库，解锁AI应用新玩法

聚焦“写作效率、视觉美观与运行性能”三方面进行全面升级，为您提供更高效、稳定的创作环境

社区富文本&Markdown编辑器全新改版上线，欢迎大家体验!

诚挚邀请您参与本次调研，分享您的真实使用感受与建议。您的反馈至关重要，感谢您的支持与参与！

社区新版编辑器体验调研

 有人设法用TensorBoard记录了损失吗？我正在使用trax ml库。我得到了这个错误TypeError：'SummaryWriter‘object is not callable。 我正在使用jaxboard中的SummaryWriter，然后将其添加到training.Loop中的callbacks中。 运行训练循环时出现错误：

问使用Trax的TensorBoard
EN

回答 1

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问使用Trax的TensorBoardEN