From 0143ad1d1b4050ece13f69008ca21e142da10535 Mon Sep 17 00:00:00 2001 From: fg-mindee Date: Tue, 13 Jul 2021 15:13:20 +0200 Subject: [PATCH] fix: Fixed typo --- references/detection/train_pytorch.py | 4 ++-- references/recognition/train_pytorch.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/references/detection/train_pytorch.py b/references/detection/train_pytorch.py index 2c73c9d1c7..7569eb202e 100644 --- a/references/detection/train_pytorch.py +++ b/references/detection/train_pytorch.py @@ -176,9 +176,9 @@ def main(args): betas=(0.95, 0.99), eps=1e-6, weight_decay=args.weight_decay) # Scheduler if args.sched == 'cosine': - scheduler = CosineAnnealingLR(optimizer, args.epochs * len(self.train_loader), eta_min=args.lr / 25e4) + scheduler = CosineAnnealingLR(optimizer, args.epochs * len(train_loader), eta_min=args.lr / 25e4) elif args.sched == 'onecycle': - scheduler = OneCycleLR(optimizer, args.lr, args.epochs * len(self.train_loader)) + scheduler = OneCycleLR(optimizer, args.lr, args.epochs * len(train_loader)) # Training monitoring current_time = datetime.datetime.now().strftime("%Y%m%d-%H%M%S") diff --git a/references/recognition/train_pytorch.py b/references/recognition/train_pytorch.py index b37dc9ca7b..fcbf77355b 100644 --- a/references/recognition/train_pytorch.py +++ b/references/recognition/train_pytorch.py @@ -166,9 +166,9 @@ def main(args): betas=(0.95, 0.99), eps=1e-6, weight_decay=args.weight_decay) # Scheduler if args.sched == 'cosine': - scheduler = CosineAnnealingLR(optimizer, args.epochs * len(self.train_loader), eta_min=args.lr / 25e4) + scheduler = CosineAnnealingLR(optimizer, args.epochs * len(train_loader), eta_min=args.lr / 25e4) elif args.sched == 'onecycle': - scheduler = OneCycleLR(optimizer, args.lr, args.epochs * len(self.train_loader)) + scheduler = OneCycleLR(optimizer, args.lr, args.epochs * len(train_loader)) # Training monitoring current_time = datetime.datetime.now().strftime("%Y%m%d-%H%M%S")