API Changes to NNDataSet. Added a few accessors to NNNetwork.

amazon-archives / amazon-dsstne

Deep Scalable Sparse Tensor Network Engine (DSSTNE) is an Amazon developed library for building Deep Learning (DL) machine learning (ML) models

Apache License 2.0

4.41k stars 731 forks source link

API Changes to NNDataSet. Added a few accessors to NNNetwork. #201

Closed kiukA9 closed 5 years ago

kiukA9 commented 5 years ago

Note Make sure to "ignore whitespace" when doing the review since I cleaned up some trailing whitespaces and spacing inconsistencies.

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.

slegrandA9 commented 5 years ago

So why the GetLayersForDataSet API call? You'd have to be kind of a dimwit to not know the name of your input or output layer. Also, ultimately a dataset will be possible to use across multiple layers to save memory. You can almost do this now, but I want to extend it further for mixture of experts networks that all train at the same time.