a76c41d2432058c33f3bd34d962ffb232cfce99a,tflearn/helpers/trainer.py,Trainer,__init__,#Trainer#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,64

Before Change

                max_to_keep=max_checkpoints,
                keep_checkpoint_every_n_hours=keep_checkpoint_every_n_hours)
            // Saver for restoring a model (With exclude variable list)
            all_vars = tf.get_collection(tf.GraphKeys.VARIABLES)
            excl_vars = tf.get_collection(tf.GraphKeys.EXCL_RESTORE_VARS)
            to_restore = [item for item in all_vars
                          if check_restore_tensor(item, excl_vars)]
            self.restorer = tf.train.Saver(

After Change

            if not self.restored:
                // TF 0.12 fix
                try:
                    init = tf.global_variables_initializer()
                except Exception:
                    init = tf.initialize_all_variables()
                self.session.run(init)

    def fit(self, feed_dicts, n_epoch=10, val_feed_dicts=None, show_metric=False,
            snapshot_step=None, snapshot_epoch=True, shuffle_all=None,

Instances