wkcn / MobulaOP

A Simple & Flexible Cross Framework Operators Toolkit
MIT License
164 stars 21 forks source link

The crash when training model on multiple GPUs #62

Closed wkcn closed 4 years ago

wkcn commented 4 years ago

Hi, there.

I found MobulaOP will crash when training model on multiple GPUs. I'm trying to fix it.

wkcn commented 4 years ago

Close it since the bug has been fixed.