Improve convert to torch script (load both training and testing)
Improve convert to torch script (load both training and testing)

Conchylicultor/DeepLearningOnGraph,Conchylicultor/DeepLearningOnGraph,Conchylicultor/DeepLearningOnGraph