927e3e63b91ea90d33171c0858af72068e503806,dtb/trainers/AutoencoderTrainer.py,AutoencoderTrainer,train,#AutoencoderTrainer#Any#Any#Any#Any#Any#,23

Before Change

                // Wait for threads to finish.
                coord.join(threads)

            return evaluator.stats(
                paths["best"], model, dataset, batch_size=args["batch_size"])

After Change

                // Wait for threads to finish.
                coord.join(threads)

            stats = evaluator.stats(
                paths["best"], model, dataset, batch_size=args["batch_size"])
            model.save({
                "args": args,
                "paths": paths,
                "steps": steps,
                "stats": stats
            })
            return model.info

Instances