IBM / mi-prometheus

Enabling reproducible Machine Learning research
http://mi-prometheus.rtfd.io/
Apache License 2.0
42 stars 18 forks source link

Numpy > 1.11.0 causes errors due to numpy.float64 not being automatically converted to Int #69

Closed sesevgen closed 5 years ago

sesevgen commented 5 years ago

Describe the bug Numpy floats do not automatically get converted to Ints when needed with later versions of numpy. Thus, pytorch network configurations that expect Ints throw errors.

To Reproduce Steps to reproduce the behavior: Run the following with a later version of numpy installed: $ mip-offline-trainer --c mi-prometheus/configs/vision/simplecnn_mnist.yaml

Expected behavior Network definitions proceed without errors

Desktop (please complete the following information):

vmarois commented 5 years ago

Adressed in #70