[Learning] Add tests

• Add unit tests for the CI checking that the gradient propagated is good.

• Carry out a full learning test with MobileNetv1 and Resnet18 on ImageNet. Create scripts and show the learning curve.