-
-
Notifications
You must be signed in to change notification settings - Fork 692
Replace mp.spawn by mp.start_processes for native comp model #1258
Copy link
Copy link
Closed
Labels
Description
🚀 Feature
As torch native mp.spawn does not work on Colab with start_method='spawn', but seem to work with start_method='fork' if using mp.start_processes. Idea is to simply replace mp.spawn by mp.start_processes here :
ignite/ignite/distributed/comp_models/native.py
Lines 282 to 285 in c667371
| mp.spawn( | |
| _NativeDistModel._dist_worker_task_fn, | |
| nprocs=nproc_per_node, | |
| args=( |
Reactions are currently unavailable