https://github.com/NVIDIA/apex/pull/1282/files/01802f623c9b54199566871b49f94b2d07c3f047
Handle len(cached_x.grad_fn.next_functions) == 1 in cached_cast by jiafatom · Pull Request #1282 · NVIDIA/apex
Using apex for mix precision training, and find one case when if x.requires_grad and cached_x.requires_grad:, the tuple cached_x.grad_fn.next_functions contains only one element. In this case, we s...
github.com
'컴퓨터 이야기~ > 소프트웨어' 카테고리의 다른 글
| js 로만 이루어진 captcha 인듯 (0) | 2024.06.06 |
|---|---|
| Android Platform Guide (0) | 2024.06.06 |
| [Cordova] ios privacy manifest 를 위한 config.xml (0) | 2024.05.17 |
| [Cordova] android adaptive icon 를 위한 config.xml (0) | 2024.05.17 |
| 임시 파일 삭제 프로그램 (0) | 2024.05.14 |
