modify

Author: neuralchen

README.md

@@ -10,6 +10,29 @@ This is an unofficial PyTorch implementation of [sngan_projection](https://githu
 - [torchviz](https://github.com/szagoruyko/pytorchviz) pip install torchviz and [graphviz](http://www.graphviz.org/) sudo apt-get install graphviz
 
 # Usage:
+There are two to run the training script:
+- Run the script directly (We recommend this way): `python3 main.py` or `python main.py`.
+    In this way, the training parameters can be modified by modifying the `parameter.py` parameter defaults.
+
+# Parameters
+|  Parameters   | Function  |
+|  :----  | :----  |
+| --version  | Experiment name |
+| --train  | Set the model stage, Ture---training stage; False---testing stage |
+| --experiment_description  | Descriptive text for this experiment  |
+
+| --total_step  | Totally training step |
+| --batch_size  | Batch size |
+| --g_lr  | Learning rate of generator |
+| --d_lr  | Learning rate of discriminator |
+
+| --parallel  | Enable the parallel training |
+| --dataset  | Set the dataset name,lsun,celeb,cifar10 |
+| --cuda  | Set GPU device number |
+| --image_path  | The root dir to training dataset |
+| --FID_mean_cov  | The root dir to dataset moments npz file |
+
+
 
 # Acknowledgement
 - [sngan_projection](https://github.com/pfnet-research/sngan_projection)

parameter.py

@@ -55,7 +55,6 @@ def getParameters():
     parser.add_argument('--parallel', type=str2bool, default=False)
     parser.add_argument('--GPUs', type=str, default='0', help='gpuids eg: 0,1,2,3  --parallel True  ')
     parser.add_argument('--dataset', type=str, default='cifar10', choices=['lsun', 'celeba','cifar10'])
-    parser.add_argument('--use_tensorboard', type=str2bool, default=True)
 
     # Path
     parser.add_argument('--image_path', type=str, default='D:\\Workspace\\SAGAN_WS\\data')