Fix comments and logger name in validate.py (#1061)

Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/1061

Differential Revision: D20238796

Pulled By: myleott

fbshipit-source-id: cf48bd7f6cdae05e91868a9d2efd91dc8e72bb12
This commit is contained in:
Myle Ott 2020-03-03 19:55:31 -08:00 committed by Facebook Github Bot
parent 6a4ad3327a
commit 077c351d7e

View File

@ -19,7 +19,7 @@ logging.basicConfig(
level=logging.INFO,
stream=sys.stdout,
)
logger = logging.getLogger('fairseq_cli.train')
logger = logging.getLogger('fairseq_cli.validate')
def main(args, override_args=None):
@ -59,7 +59,6 @@ def main(args, override_args=None):
criterion = task.build_criterion(model_args)
criterion.eval()
# Load valid dataset (we load training data below, based on the latest checkpoint)
for subset in args.valid_subset.split(','):
try:
task.load_dataset(subset, combine=False, epoch=0)