본문 바로가기

Ai/pytorch5

timm model list bat_resnext26ts beit_base_patch16_224 beit_base_patch16_384 beit_large_patch16_224 beit_large_patch16_384 beit_large_patch16_512 beitv2_base_patch16_224 beitv2_large_patch16_224 botnet26t_256 botnet50ts_256 caformer_b36 caformer_m36 caformer_s18 caformer_s36 cait_m36_384 cait_m48_448 cait_s24_224 cait_s24_384 cait_s36_384 cait_xs24_384 cait_xxs24_224 cait_xxs24_384 cait_xxs36_224 cait_xxs36_384 .. 2023. 12. 20.
cuda 버전 확인, cuda 설치 윈도우 실행 창에서 (win + R) cmd 창을 열고 nvcc --version 을 쳐준다 https://pytorch.kr/get-started/locally/ 파이토치 한국 사용자 모임 (PyTorch Korea User Group) 파이토치 한국 사용자 모임에 오신 것을 환영합니다. 딥러닝 프레임워크인 파이토치(PyTorch)를 사용하는 한국어 사용자들을 위해 문서를 번역하고 정보를 공유하고 있습니다. pytorch.kr 그리고 여기서 설치 코드를 받아온다 2023. 11. 21.
괴랄한 torch.scatter( ) 이해하기 scatter를 이해하기 위해 https://pytorch.org/docs/stable/generated/torch.Tensor.scatter_.html#torch.Tensor.scatter_ torch.Tensor.scatter_ — PyTorch 2.1 documentation Shortcuts pytorch.org 여기에 들어가보면 self[index[i][j][k]][j][k] = src[i][j][k] # if dim == 0 self[i][index[i][j][k]][k] = src[i][j][k] # if dim == 1 self[i][j][index[i][j][k]] = src[i][j][k] # if dim == 2 이렇게 친절하게 알려준다..... 친절하게 예시도 준다...(못알아 먹겠.. 2023. 11. 14.
torch.swapdims(x,0,1) 의 이해 보이는 그대로 ! dim = 0 과 dim = 1 바꿔준다 2023. 11. 14.
괴랄한 torch.gather 명령어 이해 a = torch.arange(64).view(4,4,4) print(a.shape) print(a) indi = torch.tensor([[[0,1,2,3],[0,1,2,3]],[[0,1,2,3],[0,1,2,3]],[[0,1,2,3],[0,1,2,3]]]) out = torch.gather(a,0,indi) # axis = 0 print(out) out = torch.gather(a,1,indi) # axis = 1 print(out) out = torch.gather(a,2,indi) # axis = 2 print(out) axis= 0, 1, 2 의 경우를 관찰 ! 여기서 indices는 2x4x3 (row col channel)으로 만들었는데 (torch.shape는 3x2x4) input이 i.. 2023. 11. 13.
728x90
반응형

티스토리툴바