Closed francescodisalvo05 closed 3 years ago
Replaced amp with torch.cuda.amp and removed abn and GPU parallelization, because with Colab we're not allowed to use multiple GPUs
Replaced amp with torch.cuda.amp and removed abn and GPU parallelization, because with Colab we're not allowed to use multiple GPUs