ustc_imcl_fj的博客

TensorboardXTensorboardX是一个用于可视化的模块，在pytorch中用的比较多。其支持以下类型的总结scalarimagehistogramaudiotextgraphonnx_graphembeddingpr_curveSummaryWriter提供接口，实现以下功能：在给定的目录下，创建事件文件和向事件文件中添加汇总数据事件...

torchtorch.nntorch.nn.functionaltorch.optimtorch.autograd

gathertorch.gather(input, dim, index, out=None) → Tensor沿给定轴dim, 将输入索引张量index指定位置的值进行聚合，即，指定维度的大小会发生变化。对于一个3维张量，输出可以定义为：out[i][j][k] = tensor[index[i][j][k]][j][k] # dim=0out[i][j][k] = tens...

init model parameter example

load and save model

这里写自定义目录标题PytorchPytorch‘‘RuntimeError: grad can be implicitly created only for scalar outputs’’表明你的loss不是标量