Switch model and module members to protected

This commit is contained in:
2024-05-19 20:21:54 +02:00
parent 33d4a43dca
commit 870b121c2a
2 changed files with 2 additions and 2 deletions

View File

@@ -39,7 +39,7 @@ class Model {
bool validate(); bool validate();
private: protected:
Layers::Input* inputLayer; Layers::Input* inputLayer;
Layers::Output* outputLayer; Layers::Output* outputLayer;

View File

@@ -23,7 +23,7 @@ class Module : public Layers::SequentialLayer {
const std::unordered_map<std::string, Layers::SequentialLayer*>& getLayers() const; const std::unordered_map<std::string, Layers::SequentialLayer*>& getLayers() const;
private: protected:
int inputSize; int inputSize;
int inputChannels; int inputChannels;