updated README with visualization updates

Author: kumar-shridhar

README.md

@@ -69,37 +69,27 @@ Currently, following datasets and models are supported.
 #### Bayesian
 
 `python main_bayesian.py`
-* set hyperparameters in `config_bayesian.py`
+  * set hyperparameters in `config_bayesian.py`
 
 
 #### Frequentist
 
 `python main_frequentist.py`
-* set hyperparameters in `config_frequentist.py`
-
----------------------------------------------------------------------------------------------------------
-
-
-
-### Directory Structure:
-`layers/`:  Contains ModuleWrapper, FlattenLayer, Bayesian layers (BBBConv2d and BBBLinear).  
-`models/BayesianModels/`: Contains standard Bayesian models (BBBLeNet, BBBAlexNet, BBB3Conv3FC).  
-`models/NonBayesianModels/`: Contains standard Non-Bayesian models (LeNet, AlexNet).  
-`checkpoints/`: Checkpoint directory for the best model will be saved here.  
-`tests/`: Basic unittest cases for layers and models.  
-`main_bayesian.py`: Train and Evaluate Bayesian models.  
-`config_bayesian.py`: Hyperparameters for `main_bayesian` file.  
-`main_frequentist.py`: Train and Evaluate non-Bayesian (Frequentist) models.  
-`config_frequentist.py`: Hyperparameters for `main_frequentist` file.  
-`visualize_mean_var.py`: Plotting Distributions and Line graphs of mean and variances.
+  * set hyperparameters in `config_frequentist.py`
 
 ---------------------------------------------------------------------------------------------------------
 
 
 
 ### Recording Mean and Variance:
-If `record_mean_var` is `True`, then mean and variances for layers in `record_layers` list will be logged in checkpoints directory. You can also specify recording frequency per epoch. All these mentioned parameters can be modified in `config_bayesian.py`.  
+If `record_mean_var` is `True`, then mean and variances for layers in `record_layers` list will be logged in checkpoints directory. Recording frequency per epoch can be defined. All these mentioned parameters can be modified in the `config_bayesian.py` file.  
 
+#### DistPlots
+![Average Mean value change of FC3 Layer (Node 0)](experiments/figures/fc3-node_0-mean-distplot.gif)
+![Average Standard Deviation value change of FC3 Layer (Node 0)](experiments/figures/fc3-node_0-std-distplot.gif)
+![Distribution update of FC3 Layer (Node 0)](experiments/figures/fc3-node_0-both-distplot.gif)
+
+#### LinePlots
 ![Mean value change of FC3 Layer (Node 0)](experiments/figures/fc3-node_0-mean-lineplot.jpg)
 ![Standard Deviation value change of FC3 Layer (Node 0)](experiments/figures/fc3-node_0-std-lineplot.jpg)
 
@@ -111,18 +101,32 @@ If `record_mean_var` is `True`, then mean and variances for layers in `record_la
 3. Choosing `recording_freq_per_epoch` higher than number of training iterations will raise `ZeroDivisionError`.
 
 In order to visualize the recorded values, `visualize_mean_var.py` contains `draw_distributions` and `draw_lineplot` methods. Following are the arguments which needs to be passed to `visualize_mean_var.py`:  
-1. `--filename`: Path to log file.
-2. `--data_type`: Draw plots for what? `mean` or `std`? Default is `'mean'`.  
-3. `--node_no`: Index of the node for which to draw plots. Index is after flattening of the layer. Default is 0 i.e, first node.
-4. `--plot_type`: Which plot to draw? Currently we support lineplot and distplot. Default is `'lineplot'`.
-5. `--plot_time`: Pauses the plot for this much amount of time before updating it. Default is 1 second.
-6. `--save_plots`: Whether to save plots or not. Default is 0 for No. 1 is for Yes.
-7. `--save_dir`: Directory for saving plots. Must end with `'/'`. If not provided, default directory will be `filename_directory/plots/`.
+  * `--filename`: Path to log file.
+  * `--data_type`: Draw plots for what? `mean`, `std` or `both`? Default is `'mean'`.  
+  * `--node_no`: Index of the node for which to draw plots. Index is after flattening of the layer. Default is 0 i.e, first node.
+  * `--plot_type`: Which plot to draw? Currently we support lineplot and distplot. Default is `'lineplot'`.
+  * `--plot_time`: Pauses the plot for this much amount of time before updating it. Default is 1 second.
+  * `--save_plots`: Whether to save plots or not. Default is 0 for No. 1 is for Yes.
+  * `--save_dir`: Directory for saving plots. Must end with `'/'`. If not provided, default directory will be `filename_directory/plots/`.
+
+---------------------------------------------------------------------------------------------------------
+
+### Directory Structure:
+`layers/`:  Contains ModuleWrapper, FlattenLayer, Bayesian layers (BBBConv2d and BBBLinear).  
+`models/BayesianModels/`: Contains standard Bayesian models (BBBLeNet, BBBAlexNet, BBB3Conv3FC).  
+`models/NonBayesianModels/`: Contains standard Non-Bayesian models (LeNet, AlexNet).  
+`checkpoints/`: Checkpoint directory for the best model will be saved here.  
+`tests/`: Basic unittest cases for layers and models.  
+`main_bayesian.py`: Train and Evaluate Bayesian models.  
+`config_bayesian.py`: Hyperparameters for `main_bayesian` file.  
+`main_frequentist.py`: Train and Evaluate non-Bayesian (Frequentist) models.  
+`config_frequentist.py`: Hyperparameters for `main_frequentist` file.  
+`visualize_mean_var.py`: Plotting Distributions and Line graphs of mean and variances.
 
 ---------------------------------------------------------------------------------------------------------
 
 
-If you are using this work, please cite:
+If you are using this work, please cite the authors:
 
 ```
 @article{shridhar2019comprehensive,