diff --git a/README.rst b/README.rst index e15c9c1..6ff1525 100644 --- a/README.rst +++ b/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