songqian comments

Repositories
Issues
Comments

Results 1 comments of


                                            songqian

MNIST Hogwild on Apple Silicon

same problem and I sloved by: 1. keep all multi-process processing job/function/param in CPU 2. move model/param/data to MPS in multi-process processing job/function by: model.to(torch.device("mps")) in_tensor.to(torch.device("mps")) Mostly, multi-process communication needs...