search

mlpack / models

models built with mlpack
https://models.mlpack.org/docs
BSD 3-Clause "New" or "Revised" License
33 stars 42 forks source link

VGG Implementation #78

Closed shubham1206agra closed 1 year ago

shubham1206agra commented 1 year ago

Implemented VGG using the new ann-vtable structure as a Layer for more straightforward serialization and use.

rcurtin commented 1 year ago

Ah, oops, I forgot, we also talked about adding some simple tests:

It would be nice to add some simple tests too, so that we know that VGG compiles and runs. Of course these would have to be very simple tests (like making sure that we can do a forward pass on one point, and a backwards pass on one point, and that the output is a reasonable size), so that the test does not take five years to run. :)

rcurtin commented 1 year ago

Awesome work @shubham1206agra! :+1: Just two more to merge :)