Forráskód Böngészése

Update PyTorch GAN zoo README

Jim Madge 3 éve
szülő
commit
66663e953e
1 módosított fájl, 7 hozzáadás és 13 törlés
  1. 7 13
      workflows/pytorch_GAN_zoo/README.md

+ 7 - 13
workflows/pytorch_GAN_zoo/README.md

@@ -4,8 +4,8 @@ This example builds a singularity container for [Facebook Research's PyTorch GAN
 Zoo](https://github.com/facebookresearch/pytorch_GAN_zoo).
 Zoo](https://github.com/facebookresearch/pytorch_GAN_zoo).
 
 
 The singularity container will allow you to call all the scripts from the
 The singularity container will allow you to call all the scripts from the
-project and includes are requirements. The container supports CUDA versions
-10.1, 10.2 and 11.1 on the host.
+project and includes are requirements. The container supports CUDA version 11.1
+on the host.
 
 
 ## Building
 ## Building
 
 
@@ -36,12 +36,8 @@ singularity exec pytorch_GAN_zoo.sif eval.py
 Any flags or command line arguments can be declared after the script name.
 Any flags or command line arguments can be declared after the script name.
 
 
 When training, you will need to supply the `--nv` flag to singularity so that
 When training, you will need to supply the `--nv` flag to singularity so that
-the host GPU may be used. You will also need to select a singularity app, using
-the `--app` flag to select the appropriate CUDA version. The available apps are
-`cu101`, `cu102`, and `cu111` for CUDA 10.1, 10.2 and 11.1 respectively.
+the host GPU may be used.
 
 
-For example, to pre-process the dtd dataset and train a PGAN model on a host
-with CUDA 10.2 you could run the following commands.
 
 
 ```bash
 ```bash
 singularity exec --app cu102 pytorch_GAN_zoo.sif datasets.py dtd <path to dtd dataset>/images/
 singularity exec --app cu102 pytorch_GAN_zoo.sif datasets.py dtd <path to dtd dataset>/images/
@@ -60,16 +56,14 @@ In each example the `--restart` flag is used so that checkpoints are
 periodically written during the training. The `--no_vis` flag is used to disable
 periodically written during the training. The `--no_vis` flag is used to disable
 visdom visualisations.
 visdom visualisations.
 
 
-As above, these examples assume the host has CUDA 10.2 installed.
-
 #### DTD
 #### DTD
 
 
 The DTD dataset requires no preprocessing, so the datasets script simply creates
 The DTD dataset requires no preprocessing, so the datasets script simply creates
 a configuration file.
 a configuration file.
 
 
 ```bash
 ```bash
-singularity exec --app cu102 pytorch_GAN_zoo.sif datasets.py dtd <path to dtd>/images
-singularity exec --nv --app cu102 pytorch_GAN_zoo.sif train.py PGAN -c config_dtd.json --restart --no_vis -n dtd
+singularity exec pytorch_GAN_zoo.sif datasets.py dtd <path to dtd>/images
+singularity exec pytorch_GAN_zoo.sif train.py PGAN -c config_dtd.json --restart --no_vis -n dtd
 ```
 ```
 
 
 Where `<path to dtd>` is the path of the directory extracted from the dtd
 Where `<path to dtd>` is the path of the directory extracted from the dtd
@@ -82,8 +76,8 @@ A processed dataset will be written to a directory delcared using the `-o` flag,
 `cifar-10` n this example.
 `cifar-10` n this example.
 
 
 ```bash
 ```bash
-singularity exec --app cu102 pytorch_GAN_zoo.sif datasets.py cifar10 <path to cifar-10> -o cifar10
-singularity exec --nv --app cu102 pytorch_GAN_zoo.sif train.py -c config_cifar10.json --restart --no_vis -n cifar10
+singularity exec pytorch_GAN_zoo.sif datasets.py cifar10 <path to cifar-10> -o cifar10
+singularity exec pytorch_GAN_zoo.sif train.py -c config_cifar10.json --restart --no_vis -n cifar10
 ```
 ```
 
 
 Where `<path to cifar-10>` is the path of the directory containing the pickle
 Where `<path to cifar-10>` is the path of the directory containing the pickle