lalitjain99 opened a new issue, #975: URL: https://github.com/apache/singa/issues/975
I am trying to implement distribute training using DistOpt but getting error sgd = opt.SGD(lr=0.005, momentum=0.9, weight_decay=1e-5) sgd = opt.DistOpt(sgd) model.set_optimizer(sgd) dev = device.create_cuda_gpu_on(sgd.local_rank) --------------------------------------------------------------------------- AttributeError Traceback (most recent call last) [<ipython-input-8-d0c92a72565d>](https://localhost:8080/#) in <module>() 1 sgd = opt.SGD(lr=0.005, momentum=0.9, weight_decay=1e-5) ----> 2 sgd = opt.DistOpt(sgd) 3 model.set_optimizer(sgd) 4 dev = device.create_cuda_gpu_on(sgd.local_rank) [/usr/local/lib/python3.7/dist-packages/singa/opt.py](https://localhost:8080/#) in __init__(self, opt, nccl_id, local_rank, world_size, buffSize) 723 if nccl_id is None: 724 # constructure for application using MPI --> 725 self.communicator = singa.Communicator(buffSize) 726 else: 727 # constructor for application using python multi-process module AttributeError: module 'singa.singa_wrap' has no attribute 'Communicator' -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
