search

PaulKMandal / FedCL

Federated Deep Learning in C and OpenCL
Other
1 stars 0 forks source link

Fix forward to return an array of floats. #5

Open PaulKMandal opened 11 months ago

PaulKMandal commented 11 months ago
  1. Forward should return an array of floats.
  2. The return dimension of forward needs to vary based off of the layer type (e.g. 1 for Dense, 2 for CNNs, etc.)