- 打印的时候将省略的tensor全部展开显示: torch.set_printoptions(threshold=np.inf)
- 优化器:
optimizer = torch.optim.Adam([*i3d.parameters()] + [*evaluator.parameters()],lr=args
optimizer = torch.optim.Adam([*i3d.parameters()] + [*evaluator.parameters()],lr=args