issues
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
Forward should return an array of floats.
The return dimension of forward needs to vary based off of the layer type (e.g. 1 for Dense, 2 for CNNs, etc.)