From 139488e0afd28f1460f8295219175949cdd32603 Mon Sep 17 00:00:00 2001
From: ncguilbeault <ncguilbeault@gmail.com>
Date: Wed, 29 Jan 2025 19:07:00 +0000
Subject: [PATCH] Modified torch module classes to be internel

---
 src/Bonsai.ML.Torch/NeuralNets/Models/AlexNet.cs   | 2 +-
 src/Bonsai.ML.Torch/NeuralNets/Models/MNIST.cs     | 2 +-
 src/Bonsai.ML.Torch/NeuralNets/Models/MobileNet.cs | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/Bonsai.ML.Torch/NeuralNets/Models/AlexNet.cs b/src/Bonsai.ML.Torch/NeuralNets/Models/AlexNet.cs
index c80a3d50..c3d19d55 100644
--- a/src/Bonsai.ML.Torch/NeuralNets/Models/AlexNet.cs
+++ b/src/Bonsai.ML.Torch/NeuralNets/Models/AlexNet.cs
@@ -7,7 +7,7 @@ namespace Bonsai.ML.Torch.NeuralNets.Models
     /// <summary>
     /// Modified version of original AlexNet to fix CIFAR10 32x32 images.
     /// </summary>
-    public class AlexNet : Module<Tensor, Tensor>
+    internal class AlexNet : Module<Tensor, Tensor>
     {
         private readonly Module<Tensor, Tensor> features;
         private readonly Module<Tensor, Tensor> avgPool;
diff --git a/src/Bonsai.ML.Torch/NeuralNets/Models/MNIST.cs b/src/Bonsai.ML.Torch/NeuralNets/Models/MNIST.cs
index 8a5f84db..8bd3e0a4 100644
--- a/src/Bonsai.ML.Torch/NeuralNets/Models/MNIST.cs
+++ b/src/Bonsai.ML.Torch/NeuralNets/Models/MNIST.cs
@@ -7,7 +7,7 @@ namespace Bonsai.ML.Torch.NeuralNets.Models
     /// <summary>
     /// Represents a simple convolutional neural network for the MNIST dataset.
     /// </summary>
-    public class MNIST : Module<Tensor,Tensor>
+    internal class MNIST : Module<Tensor,Tensor>
     {
         private readonly Module<Tensor, Tensor> conv1;
         private readonly Module<Tensor, Tensor> conv2;
diff --git a/src/Bonsai.ML.Torch/NeuralNets/Models/MobileNet.cs b/src/Bonsai.ML.Torch/NeuralNets/Models/MobileNet.cs
index 0faa1062..a5f7701a 100644
--- a/src/Bonsai.ML.Torch/NeuralNets/Models/MobileNet.cs
+++ b/src/Bonsai.ML.Torch/NeuralNets/Models/MobileNet.cs
@@ -9,7 +9,7 @@ namespace Bonsai.ML.Torch.NeuralNets.Models
     /// <summary>
     /// MobileNet model.
     /// </summary>
-    public class MobileNet : Module<Tensor, Tensor>
+    internal class MobileNet : Module<Tensor, Tensor>
     {
         private readonly long[] planes = [ 64, 128, 128, 256, 256, 512, 512, 512, 512, 512, 512, 1024, 1024 ];
         private readonly long[] strides = [ 1, 2, 1, 2, 1, 2, 1, 1, 1, 1, 1, 2, 1 ];