From 149628a0031f595e6c31a5e6ce6797657c96fd16 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Sat, 5 Oct 2019 15:02:46 -0400 Subject: [PATCH] cleaning up demos --- examples/basic_examples/lightning_module_template.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/basic_examples/lightning_module_template.py b/examples/basic_examples/lightning_module_template.py index 7188d804..f032e9a0 100644 --- a/examples/basic_examples/lightning_module_template.py +++ b/examples/basic_examples/lightning_module_template.py @@ -8,7 +8,7 @@ from torchvision.datasets import MNIST import torchvision.transforms as transforms import torch import torch.nn.functional as F -from test_tube import HyperOptArgumentParser +from argparse import ArgumentParser from torch import optim from torch.utils.data import DataLoader from torch.utils.data.distributed import DistributedSampler @@ -231,7 +231,7 @@ class LightningTemplateModel(LightningModule): :param root_dir: :return: """ - parser = HyperOptArgumentParser(strategy=parent_parser.strategy, parents=[parent_parser]) + parser = ArgumentParser(parents=[parent_parser]) # param overwrites # parser.set_defaults(gradient_clip_val=5.0)