How to train with two GPUs? - Githubissues

Zhangjinso / PISE

123 stars 28 forks source link

How to train with two GPUs? #8

Closed Ha0Tang closed 3 years ago

Zhangjinso commented 3 years ago

Our code is based on GFLA, so you can refer to https://github.com/RenYurui/Global-Flow-Local-Attention/issues/22.