import tensorflow as tf

        self.generator = generator
        self.model_pred = importlib.import_module(self.flags.model) # import network module
                
        self.use_trans_loss = self.flags.use_transformation_loss
        self.use_input_trans = self.flags.use_input_transform
        self.use_feature_trans = self.flags.use_feature_transform
    
        self.is_training_pl = tf.placeholder(tf.bool, shape=(), name='is_training_pl')

        self.bn_decay = train_rotation_prediction.get_bn_decay(batch)

        self.get_pred = partial(self.model_pred.get_model, 
                                is_training=self.is_training_pl, 
                                bn_decay=self.bn_decay,
