Use torch.accelerator API in VAE examples, fix README

Author: eromomon

vae/README.md

@@ -8,13 +8,12 @@ pip install -r requirements.txt
 python main.py
 ```
 
-The main.py script accepts the following arguments:
+The main.py script accepts the following optional arguments:
 
 ```bash
-optional arguments:
-  --batch-size		input batch size for training (default: 128)
-  --epochs		number of epochs to train (default: 10)
-  --accel		  use accelerator
-  --seed		random seed (default: 1)
-  --log-interval	how many batches to wait before logging training status
+--batch-size            input batch size for training (default: 128)
+--epochs                number of epochs to train (default: 10)
+--accel                 use accelerator
+--seed                  random seed (default: 1)
+--log-interval	        how many batches to wait before logging training status
 ```