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 optimizer is:\n{:}".format(optimizer))
|
||||||
logger.log("The objective is {:}".format(objective))
|
logger.log("The objective is {:}".format(objective))
|
||||||
logger.log("The metric is {:}".format(metric))
|
logger.log("The metric is {:}".format(metric))
|
||||||
logger.log("The iters_per_epoch = {:}, estimated epochs = {:}".format(
|
logger.log(
|
||||||
iters_per_epoch, args.steps // iters_per_epoch))
|
"The iters_per_epoch = {:}, estimated epochs = {:}".format(
|
||||||
|
iters_per_epoch, args.steps // iters_per_epoch
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
model, objective = torch.nn.DataParallel(model).cuda(), objective.cuda()
|
model, objective = torch.nn.DataParallel(model).cuda(), objective.cuda()
|
||||||
scheduler = xmisc.LRMultiplier(
|
scheduler = xmisc.LRMultiplier(
|
||||||
|
@ -1,6 +1,12 @@
|
|||||||
#####################################################
|
#####################################################
|
||||||
# Copyright (c) Xuanyi Dong [GitHub D-X-Y], 2020.06 #
|
# 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:
|
class AverageMeter:
|
||||||
|
Loading…
Reference in New Issue
Block a user