diff --git a/train.py b/train.py index c7d87643..0031f777 100644 --- a/train.py +++ b/train.py @@ -320,9 +320,6 @@ def train(cfg, return results - - - if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument('--epochs', type=int, default=100, help='number of epochs')