Method load

examples/reinforcement_learning/tutorial_C51.py:262–265

(self, path)

        tl.files.save_weights_to_hdf5(os.path.join(path, 'q_net.hdf5'), self.qnet)

    def load(self, path):
        if path is None:
            path = os.path.join('model', '_'.join([alg_name, env_id]))
        tl.files.load_hdf5_to_weights_in_order(os.path.join(path, 'q_net.hdf5'), self.qnet)

    @tf.function
    def _train_func(self, b_o, b_index, b_m):