a multi-process problem in the controller

[sfrazavi]

Thanks to the authors for sharing their code.
I executed the controller on Ubuntu 18.04 with one GPU.
Unfortunately, the program couldn't execute due to a multi-process problem with Cuda. After that, I put all of the code except functions in the main function and called it.
Now, it works properly.