Skip to content

Replace mp.spawn by mp.start_processes for native comp model #1258

@vfdev-5

Description

@vfdev-5

🚀 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 :

mp.spawn(
_NativeDistModel._dist_worker_task_fn,
nprocs=nproc_per_node,
args=(

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions