undertherain / benchmarker / 769

44%

push

travis-ci

Cosmoflow (#87)

* Add stub cosmoflow

* Add argparse to cosmoflow

* Fix arg parse of input shape

* Add model compilation

* Guess get_data()

* Refactor into build_model

* Finish cosmoflow (see full commit message!)

It seems cosmoflow is finished but it still doesn't work, because ATM
`do_tensorflow.py` does only classification (see line 77, where it
applies `to_categorical` to `y_train`).  So to make it work - you have
to comment out that line.

* Remove to_categorical()

* Change resnet, vggs, xception to use sparse CCE

* Add cosmoflow test

* Add cosmoflow/pytorch.py

* Move `proc_params()` to separate file

* Add support for channels_first

* Finish pytorch inference for cosmoflow

* Remove unused cosmoflow model

* Add `Regression` class to `helpers_torch.py`

* Make pytorch cosmoflow a regression

* Add test/pytorch/test_cosmoflow.py

Co-authored-by: Emil VATAI <vatai@x1carbon.subliminal>

128 of 450 branches covered (28.44%)

Branch coverage included in aggregate %.

1182 of 2208 relevant lines covered (53.53%)

0.54 hits per line