search

dhimasryan / MOSA-Net-Cross-Domain

MIT License
47 stars 9 forks source link

wrapped inference inside torch.no_grad #3

Closed Wataru-Nakata closed 2 years ago

Wataru-Nakata commented 2 years ago

fixing issue regarding #2