ANUJ KUMAR

icon location I am a computer vision and machine learning enthusiast. I did internship at Sony AI Japan and UCSD. I love to develop softwares that solve real world problems.

Results 2 comments of ANUJ KUMAR

Multiple GPU problems?

Since model_output file is a generator object that returns an iterable rather than the tensors. In order to retrieve all the outputs: `for i, out in enumerate(model_output.sample): print("output on gpu...

Multiple GPU problems?

Yeah maybe you should try: `model = nn.parallel.DistributedDataParallel(model)` to parallelize the forward pass of a model across multiple GPUs.