convert a pytorch model to a model for edge device - kuroko1t/nne