I need to know all the weight values，How can I output the weight of the training process？

criterion = nn.CrossEntropyLoss().cuda()


optimizer = torch.optim.SGD(model.parameters(), args.lr 20,
momentum=args.momentum,
weight_decay=args.weight_decay)


if args.evaluate:
validate(val_loader, model, criterion)


for epoch in range(args.start_epoch, args.epochs):
adjust_learning_rate(optimizer, epoch)


    # train for one epoch
train(train_loader, model, criterion, optimizer, epoch)



    # evaluate on validation set
prec1 = validate(val_loader, model, criterion)



    # remember best prec@1 and save checkpoint
is_best = prec1 > best_prec1
best_prec1 = max(prec1, best_prec1)
save_checkpoint({
    'epoch': epoch + 1,
    'state_dict': model.state_dict(),
    'best_prec1': best_prec1,
}, is_best, filename='/home/zl/models/checkpoint_{}.pth.tar'.format(epoch))