From e21141909b482109f33f588bf8bd3dbf9ec0e7f6 Mon Sep 17 00:00:00 2001 From: William Falcon <waf2107@columbia.edu> Date: Thu, 14 May 2020 13:41:05 -0400 Subject: [PATCH] updated seed --- src/single_file_mnist/README.md | 22 ---------------- src/single_file_mnist/__init__.py | 0 src/single_file_mnist/simplest_mnist.py | 34 ------------------------- 3 files changed, 56 deletions(-) delete mode 100644 src/single_file_mnist/README.md delete mode 100644 src/single_file_mnist/__init__.py delete mode 100644 src/single_file_mnist/simplest_mnist.py diff --git a/src/single_file_mnist/README.md b/src/single_file_mnist/README.md deleted file mode 100644 index 4f7b00d..0000000 --- a/src/single_file_mnist/README.md +++ /dev/null @@ -1,22 +0,0 @@ -## MNIST -In this readme, give instructions on how to run your code. - -#### CPU -```bash -python mnist_trainer.py -``` - -#### Multiple-GPUs -```bash -python mnist_trainer.py --gpus 4 -``` - -or specific GPUs -```bash -python mnist_trainer.py --gpus '0,3' -``` - -#### On multiple nodes -```bash -python mnist_trainer.py --gpus 4 --nodes 4 --precision 16 -``` diff --git a/src/single_file_mnist/__init__.py b/src/single_file_mnist/__init__.py deleted file mode 100644 index e69de29..0000000 diff --git a/src/single_file_mnist/simplest_mnist.py b/src/single_file_mnist/simplest_mnist.py deleted file mode 100644 index ac2719b..0000000 --- a/src/single_file_mnist/simplest_mnist.py +++ /dev/null @@ -1,34 +0,0 @@ -""" -This file runs the main training/val loop, etc... using Lightning Trainer -""" -from pytorch_lightning import Trainer, seed_everything -from argparse import ArgumentParser -from srv.mnist.mnist import CoolSystem - -# sets seeds for numpy, torch, etc... -# must do for DDP to work well -seed_everything(123) - -def main(args): - # init module - model = CoolSystem(hparams=args) - - # most basic trainer, uses good defaults - trainer = Trainer.from_argparse_args(args) - trainer.fit(model) - - -if __name__ == '__main__': - parser = ArgumentParser(add_help=False) - - # add args from trainer - parser = Trainer.add_argparse_args(parser) - - # give the module a chance to add own params - # good practice to define LightningModule speficic params in the module - parser = CoolSystem.add_model_specific_args(parser) - - # parse params - args = parser.parse_args() - - main(args) -- GitLab