Alright I solved it. Given that my parameters are stored inside args. the way to do it would be,

You mean you want the new task created by add_step to take in certain parameters? Provided where/by who?

Sorry for the late reply. The situation is that when I ran the task initially, it took arguments in using ArgParse. It took in a lot of arguments. Now my understanding is that add_step() clones that task. I want that to happen but I would like to be able to modify some of the values of the args, e.g epochs or some other argument.