Open cumtsunzhenxing opened 1 month ago
The predict method is called by setting gpus=2, but only one gpu is used. model.predict(samples=samples,batch_size=2, gpus=2, devices=2, accelerator="gpu", strategy="ddp")
model.predict(samples=samples,batch_size=2, gpus=2, devices=2, accelerator="gpu", strategy="ddp")
The predict method is called by setting gpus=2, but only one gpu is used.
model.predict(samples=samples,batch_size=2, gpus=2, devices=2, accelerator="gpu", strategy="ddp")