search

originrose / cortex

Machine learning in Clojure
Eclipse Public License 1.0
1.27k stars 111 forks source link

Resnet 50 Retrain example enhancements #241

Closed gigasquid closed 7 years ago

gigasquid commented 7 years ago

I finally got time to look at the RESNET50 retraining example.

Thank you! It's great and surprisingly, it actually ran an locally on my mac!

I went through the documentation and enhanced it with things I ran into. I also added some functions to test a pic and label it.

There is another function I added to continue the training epoch by epoch, but would appreciate feedback if it is correct.

Hope this helps - and keep up the great work 💯

cnuernber commented 7 years ago

@gigasquid : Let me know if I should land this PR or if you want to work on it some more. I think you are ready but I am not certain.

gigasquid commented 7 years ago

@cnuernber I'd like a little more time. I'm going to make the code changes we talked about and I'm digging into verifying the layers of the resnet50 . I'll ping you when it is ready for another look. Thanks 😸

gigasquid commented 7 years ago

@cnuernber @harold I updated the PR to not use snapshots and reverted the change in the resize_images.sh. It's ready to go after you review.