tests |
---|
David Silva-Sánchez, Lars Dingeldein, Roberto Covino, Pilar Cossio
- Lampe
- SciPy
- NumPy
- PyTorch
- json
https://github.com/DSilva27/cryo_em_SBI.git
python3 -m pip install .
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_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