minor fixes

Author: kumar-shridhar

README.md

@@ -71,7 +71,7 @@ Currently, following datasets and models are supported.
 * set hyperparameters in `config_bayesian.py`
 
 
-#### Bayesian
+#### Frequentist
 
 `python main_frequentist.py`
 * set hyperparameters in `config_frequentist.py`
@@ -108,4 +108,4 @@ If you are using this work, please cite:
 }
 ```
 
---------------------------------------------------------------------------------------------------------
+--------------------------------------------------------------------------------------------------------

config_bayesian.py

@@ -1,5 +1,5 @@
 ############### Configuration file for Bayesian ###############
-n_epochs = 150
+n_epochs = 50
 lr_start = 0.001
 num_workers = 4
 valid_size = 0.2

config_frequentist.py

@@ -1,7 +1,6 @@
 ############### Configuration file for Frequentist ###############
-n_epochs = 150
+n_epochs = 50
 lr = 0.01
-beta_type = "Standard"
 num_workers = 4
 valid_size = 0.2
 batch_size = 256