fa0fea934bad76e9906c013e377ce80c77cbd5e4,pytext/trainers/hogwild_trainer.py,HogwildTrainer,train,#HogwildTrainer#Any#Any#Any#Any#Any#Any#Any#,35

Before Change

        for rank in range(1, self.num_workers):
            // Initialize the batches with different randome states.
            train_iter.batches.init_epoch()
            p = mp.Process(
                target=self.real_trainer.train,
                args=(
                    train_iter,
                    eval_iter,
                    model,
                    metric_reporter,
                    optimizers,
                    pytext_config,
                    scheduler,
                    None,
                    rank,
                ),
            )
            processes.append(p)
            p.start()

        training_result: List = Manager().list()  // Actual type is ListProxy.
        self.real_trainer.train(
            train_iter,

After Change

            for param in model.parameters():
                param.share_memory_()

        return super().train(
            train_iter,
            eval_iter,
            model,
            metric_reporter,
            optimizers,
            pytext_config,
            scheduler,
        )

Instances