Use only cosmology parameters in loss function

[Chris-Pedersen]

Have modified the CNNs to have variable feature output dimensions instead being hardcoded fixed to 6 or 12. Need to update the test part of the test_sn script then update the optuna scripts.