반응형
nn.Parameter(torch.Tensor(batch, h_dim))위의 tensor는 종종 nan element를 발생시킨다. 따라서 다음과 같이 tensor 생성에 사용하는것을 추천한다.
nn.Parameter(torch.rand(batch, h_dim))'Deep Learning' 카테고리의 다른 글
| Deformable DETR attention operation cuda build (0) | 2023.06.20 |
|---|---|
| Image Augmentation (Photometric) 방법 (0) | 2023.05.31 |
| Pytorch DDP (Distributed Data Parallel) 사용 관련 웹페이지 모음 (0) | 2023.05.16 |
| Pytorch에서 meshgrid 만들기 (0) | 2023.04.12 |
| Nuscenes Warping Test Code (0) | 2023.04.12 |