Closed richielo closed 5 years ago
Simply changing the shape at the flatten and linear layer would fix the error
@bhairavmehta95 can you have a look at this PR ?
Fixes #112. Thank you @richielo for doing this!
Simply changing the shape at the flatten and linear layer would fix the error