Skip to content

Gopal-Dahale/hpmoCNN

Repository files navigation

Summary

The following layers have been implemented

  1. Conv
  2. Fully connected
  3. Activation (Relu, sigmoid)
  4. Softmax
  • Using Upsampled mnist dataset.
  • Working offload/prefetch mechanism with min-heap implementation.

Results for main224

  • 5 epochs
  • 0.01 learning rate
  • 1000 train
  • 500 test
Batch Size Val_acc Test_acc
128 106 54
256 NEM NEM

NEM - Not enough memory