The code does not need to be changed. You only need to configure the environment and install the corresponding libraries to train and test.