コード例 #1
0
ファイル: AlexNet.cs プロジェクト: markusweimer/PytorchSharp
 private static void Train(
     NN.Module model,
     NN.Optimizer optimizer,
     Loss loss,
     IEnumerable <(TorchTensor, TorchTensor)> dataLoader,
コード例 #2
0
ファイル: MNIST.cs プロジェクト: cartermp/TorchSharp
 private static void Train(
     Model model,
     NN.Optimizer optimizer,
     Loss loss,
     Device device,
     IEnumerable <(TorchTensor, TorchTensor)> dataLoader,