print number of parameters of the model

reading_machine/include/Classifier.hpp

@@ -24,6 +24,7 @@ class Classifier
   TransitionSet & getTransitionSet();
   NeuralNetwork & getNN();
   const std::string & getName() const;
+  int getNbParameters() const;
 };
 
 #endif

reading_machine/src/Classifier.cpp

@@ -20,6 +20,16 @@ Classifier::Classifier(const std::string & name, std::filesystem::path path, std
   initNeuralNetwork(definition);
 }
 
+int Classifier::getNbParameters() const
+{
+  int nbParameters = 0;
+
+  for (auto & t : nn->parameters())
+    nbParameters += torch::numel(t);
+
+  return nbParameters;
+}
+
 TransitionSet & Classifier::getTransitionSet()
 {
   return *transitionSet;

trainer/src/MacaonTrain.cpp

@@ -99,6 +99,8 @@ int MacaonTrain::main()
 
   ReadingMachine machine(machinePath.string());
 
+  fmt::print(stderr, "[{}] Model has {} parameters\n", util::getTime(), util::int2HumanStr(machine.getClassifier()->getNbParameters()));
+
   BaseConfig goldConfig(mcdFile, trainTsvFile, trainRawFile);
   BaseConfig devGoldConfig(mcdFile, computeDevScore ? (devRawFile.empty() ? devTsvFile : "") : devTsvFile, devRawFile);
   SubConfig config(goldConfig, goldConfig.getNbLines());