ABOUT ME

-

Today
-
Yesterday
-
Total
-
  • pytorch2
    카테고리 없음 2022. 3. 31. 11:09

    totensor 는
    넘파이 h w c 에서 텐서 c h w 로 바뀌어 준다
    범위도 0 에서 255에서 0에서1로 바꾸어준다

    cpu 가 load 하므로 배치사이즈가클경우 num worker 가 많을수록? 좋다


    3채널을 입력으로 받아 64채널의 아웃풋
    30*5*5의 이미지를 받아 flatten시켜서 받고 128 개의 fc 아웃풋

    view 함수

    순번이 predicted 임


    채널을 이용해 학습

    인풋 weight 를 직접 넣어줘야 함


    -1자동으로 숫자 선택 (16이 고정되어있는 상태)


    3,3 대신 3,5 필터

    argmax 는 max_indices이다


Designed by Tistory.