1
2
3
4
5
6
|
import gc
obj = model
gc.collect()
torch.cuda.empty_cache() # PyTorch thing
|
cs |
실험 시 매우 유용하고 자면서 모델을 돌릴 때 필수다
'AI > 부스트 캠프' 카테고리의 다른 글
부스트 캠프 교육 정리 (0) | 2021.04.12 |
---|---|
부스트캠프 - AI Tech (0) | 2021.01.16 |
1
2
3
4
5
6
|
import gc
obj = model
gc.collect()
torch.cuda.empty_cache() # PyTorch thing
|
cs |
실험 시 매우 유용하고 자면서 모델을 돌릴 때 필수다
부스트 캠프 교육 정리 (0) | 2021.04.12 |
---|---|
부스트캠프 - AI Tech (0) | 2021.01.16 |