updated the readme

README.rst

@@ -11,15 +11,10 @@ cryo_em_SBI
       - | |githubactions|
 
 
-.. |githubactions| image:: https://github.com/DSilva27/cryo_em_SBI/actions/workflows/python-package.yml/badge.svg?branch=iss6
+.. |githubactions| image:: https://github.com/DSilva27/cryo_em_SBI/actions/workflows/python-package.yml/badge.svg?branch=cryoSBI
     :alt: Testing Status
     :target: https://github.com/DSilva27/cryo_em_SBI/actions
 
-Collaborators
--------------
-
-David Silva-Sánchez, Lars Dingeldein, Roberto Covino, Pilar Cossio
-
 Dependencies
 ------------
 
@@ -46,26 +41,17 @@ Install the module
 Using the code
 --------------
 
-Generating data from command line
-~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
-
-.. code:: bash
-
-    generate_training_data 
-        --config_file experiments/benchmark_hsp90/image_params_snr01_128.json \
-        --num_train_samples 10 \
-        --num_val_samples 10 \
-        --file_name "test1" \
-        --n_workers 24
-
-
 Train posterior from command line
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
 .. code:: bash
 
     train_npe_model \
-        --image_config_file experiments/benchmark_hsp90/image_params_training.json \
-        --train_config_file experiments/benchmark_hsp90/resnet18_encoder.json\
-        --epochs 100 \
-        --estimator_file experiments/benchmark_hsp90/posterior_hsp90.estimator \
-        --loss_file experiments/benchmark_hsp90/posterior_hsp90.loss 
+        --image_config_file image_config.json \
+        --train_config_file train_config.json\
+        --epochs 450 \
+        --estimator_file ../posterior/test.estimator \
+        --loss_file ../posterior/test.loss \
+        --n_workers 4 \
+        --train_device cuda \
+        --saving_freq 20 \
+        --simulation_batch_size 2048