Fix errors

This commit is contained in:
D-X-Y 2021-07-02 09:19:39 +00:00
parent 437f62b055
commit 9136f33684
3 changed files with 12 additions and 3 deletions

@ -1 +1 @@
Subproject commit 5e5fc2451d1702f08e2fa933107aa45b6356b6ed
Subproject commit 2d4f0e80f98211ba2e1f25a329ad2421fb8087cd

View File

@ -74,8 +74,11 @@ def main(args):
logger.log("The optimizer is:\n{:}".format(optimizer))
logger.log("The objective is {:}".format(objective))
logger.log("The metric is {:}".format(metric))
logger.log("The iters_per_epoch = {:}, estimated epochs = {:}".format(
iters_per_epoch, args.steps // iters_per_epoch))
logger.log(
"The iters_per_epoch = {:}, estimated epochs = {:}".format(
iters_per_epoch, args.steps // iters_per_epoch
)
)
model, objective = torch.nn.DataParallel(model).cuda(), objective.cuda()
scheduler = xmisc.LRMultiplier(

View File

@ -1,6 +1,12 @@
#####################################################
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2020.06 #
#####################################################
# In this python file, it contains the meter classes#
# , which may need to use PyTorch or Numpy. #
#####################################################
import abc
import torch
import torch.nn.functional as F
class AverageMeter: