in training mode (some modules like dropout modules have different
            behaviors between training and evaluation).
al