/pytorch_train

Test the traning using pytorch

Primary LanguageJupyter Notebook

pytorch_train

Test the traning using pytorch, the training model can be used in C++ for predicting.