model.to(device) for Pytorch Lighting

前端未结

关注

 0  1672

I currently train my model using GPUs using Pytorch Lightning

trainer = pl.Trainer( gpus=[0,1],  
        distributed_backend=\'ddp\', 
        resume_from_ch


                      
              相关标签: