/MobilePyTorch

Primary LanguageObjective-C++

Mobile PyTorch

MobilePyTorch builds BNNS network from pretrained PyTorch models.

  1. Create a BNNS MNIST network by hand to get familiar with BNNS interface
  2. Create a BNNSBuilder that takes in only parameters available from potential dumped python files(input/output dimensions, weights, biases, structures) to build the same network
  3. Python dumper / ObjC(or c++) loader