def set_train(layer, layerinfo=layerinfo):
            "Set layer into training mode if instance of `layerinfo`."
            if isinstance(layer, tuple(layerinfo)):
                layer.train()
Set layer into training mode if instance of `layerinfo`.
art/estimators/pytorch.py