Using curriculum learning to train the Kuka robot
The first step before actually kicking off some training is to customize the Kuka class as well as the KukaGymEnv to make them work with the curriculum learning parameters we described above. So, let's do that next.
Customizing the environment for curriculum learning
First, we start with creating a CustomKuka class which inherits the original Kuka class of PyBullet. Here is how we do it:
Chapter14/custom_kuka.py
- We first need to create the new class, and accept an additional argument,
jp_overridedictionary, which stands for joint position override.class CustomKuka(Kuka): def __init__(self, *args, jp_override=None, **kwargs): self.jp_override = jp_override super(CustomKuka, self).__init__(*args...