Question about main.py

[wngh1187]

Question about main.py

[wngh1187]

Thank you for your interest in our research.
We used pytorch's multiprocessing package to share the file system between gpus.
The "run" function in main.py called from the "spawn" function needs the process index.
Information related to this can be found at https://pytorch.org/docs/stable/_modules/torch/multiprocessing/spawn.html#spawn.

Additionally, we confirmed that training works when we run the run/triain.py file in the current code.
Do you have any problems running our code?

[wtl-lang]

Thank you, when I run the main.py, there is an error "the function run(line 37 in main.py) cannot be found in main", I move the function 'run' to another file, the code runs. Also, there are some errors about the "all_gather()", which makes the list can not be indexed. I run the code in one GPU, so I write below:
lists = all_gather(list)
lists_ = list[0]
for i in range(len(list_)):
Finally, the code runs with no errors.

[wngh1187]

Thank you for checking.
Our code is based on a multi-GPU environment.
It seems that an error occurred to you because we did not consider the single GPU environment.
First of all, you can use the code as you modified it, and I will upload a single GPU code as soon as possible.