i think its working, need to diagnose and test
WIP not sure how to initialize layers w and wo sizes