talos icon indicating copy to clipboard operation
talos copied to clipboard

Add multi_gpu support to AutoModel

Open beeb opened this issue 6 years ago • 3 comments

Everything is in the title! Would be lovely to be able to add a number of GPUs to use with AutoML / AutoModel.

Thanks!

beeb avatar Aug 16 '19 07:08 beeb

Makes total sense, will be added :)

mikkokotila avatar Aug 16 '19 14:08 mikkokotila

When you try to use keras' multi_gpu_model you might find that it gives an Exception TypeError: object of type 'NoneType' has no len() regarding model.outputs. (at least for regression tasks = continuous, did not try other tasks)

Searching for a solution on SO I found this and that.

Hope you can find a solution!

EDIT: it is not a problem for the "dense" networks, only for the other types.

beeb avatar Aug 19 '19 14:08 beeb

Oh actually I managed to fix it by specifying an input shape for all network types. That's all that was missing. See my second SO link in the reply above.

beeb avatar Aug 19 '19 15:08 beeb