lib/torch/nn/module.rb in torch-rb-0.8.1 vs lib/torch/nn/module.rb in torch-rb-0.8.2

@@ -1,9 +1,11 @@
 module Torch
   module NN
     class Module
       include Utils
 
+      attr_reader :training
+
       def initialize
         @training = true
         @parameters = {}
         @buffers = {}
         @modules = {}