Fix errors
This commit is contained in:
parent
437f62b055
commit
9136f33684
@ -1 +1 @@
|
||||
Subproject commit 5e5fc2451d1702f08e2fa933107aa45b6356b6ed
|
||||
Subproject commit 2d4f0e80f98211ba2e1f25a329ad2421fb8087cd
|
@ -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(
|
||||
|
@ -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:
|
||||
|
Loading…
Reference in New Issue
Block a user