62c7ab518214286a4721dd7410978effd6d05471,rllib/models/modelv2.py,ModelV2,from_batch,#ModelV2#Any#Any#,232

Before Change

        }
        if SampleBatch.PREV_ACTIONS in train_batch:
            input_dict["prev_actions"] = train_batch[SampleBatch.PREV_ACTIONS]
        if SampleBatch.PREV_REWARDS in train_batch:
            input_dict["prev_rewards"] = train_batch[SampleBatch.PREV_REWARDS]
        states = []
        i = 0
        while "state_in_{}".format(i) in train_batch:
            states.append(train_batch["state_in_{}".format(i)])

After Change

            states.append(train_batch["state_in_{}".format(i)])
            i += 1
        ret = self.__call__(train_batch, states, train_batch.get("seq_lens"))
        del train_batch["is_training"]
        return ret

    def import_from_h5(self, h5_file: str) -> None:
        Imports weights from an h5 file.

Instances