标签：: PyTorch

2018
0

Howto

PyTorch调参

在写CVAE模型的过程中，遇到一个loss突然变大的过程，看到网络上说由于Adam的原因，所以会导致收敛不稳定。可以把amdgrad参数打开。1optimizer = Adam(filter(lambda p: p.requires_grad, model.parameters()), lr=args.lr, weight_decay=args.wd, amsgrad=True)

2017
0

Howto

PyTorch入门

基本操作Tensors可以迁移到cuda上进行操作12345# let us run this cell only if CUDA is availableif torch.cuda.is_available(): x = x.cuda() y = y.cuda() x + y autogradpytorch所有张量的中心是autograd库。而autograd的中心组件是aut