darts icon indicating copy to clipboard operation
darts copied to clipboard

Please upadate codes for latest version of PyTorch

Open LT1st opened this issue 1 year ago • 0 comments

In file train.py and train_search.py, you need to replace this:

    # objs.update(loss.data[0], n)
    # top1.update(prec1.data[0], n)
    # top5.update(prec5.data[0], n)

to

    objs.update(loss.item(), n)
    top1.update(prec1.item(), n)
    top5.update(prec5.item(), n)

LT1st avatar Jul 15 '23 07:07 LT1st