[에러기록] assertionerror: if capturable=false, state_steps should not be cuda tensors.

2023. 2. 23. 22:45·연구 Research/인공지능 Artificial Intelligent

pytorch 안에서 adam을 쓸 때 자꾸 위와 같은 에러가 떠서 이를 위해

    optimizer = optim.Adam(func.parameters(), lr=args.lr)
    optimizer.param_groups[0]['capturable'] = True

 

다음과 같이 'capturable'을 true로 바꿔줘야했다. 그런데 다른 사람들의 라이브러리를 쓰다보면 optimizer가 라이브러리 안에 포함되어있어서 위와 같은 코드를 넣어주기 힘든 경우가 발생했다.

따라서 아래와 같이 가상환경 안에 있는 코드에 접근하여서 에러가 나는 부분을 주석처리해주었다.

이렇게 하는 이유는 실제로 위의 error가 코드 성능에 악영향을 미치지 않기 때문이다. params 및 state_steps이 CUDA tensor일 필요가 없다면 그냥 무시하는 게 낫다고 생각한다.

 

 

혹시 더 좋은 방법이 있다면 이 글을 수정하도록 하겠다.

 

저작자표시 비영리 변경금지 (새창열림)

'연구 Research > 인공지능 Artificial Intelligent' 카테고리의 다른 글

[PyTorch] 개별 파라미터 learning rate 다르게 설정 및 learning rate 확인  (2) 2023.03.05
[JAX] Windows에서도 JAX 사용하기  (0) 2023.02.24
[인공지능] Windows 기준 CUDA & cuDNN 최신 버전 설치 (CUDA 11.6 이상)  (0) 2023.02.07
[PyTorch] retain_graph = True라고 했음에도 backward 문제가 발생하는 경우  (0) 2023.01.05
[PyTorch] 모델 저장/불러오기 및 모델 수정하기  (0) 2023.01.04
'연구 Research/인공지능 Artificial Intelligent' 카테고리의 다른 글
  • [PyTorch] 개별 파라미터 learning rate 다르게 설정 및 learning rate 확인
  • [JAX] Windows에서도 JAX 사용하기
  • [인공지능] Windows 기준 CUDA & cuDNN 최신 버전 설치 (CUDA 11.6 이상)
  • [PyTorch] retain_graph = True라고 했음에도 backward 문제가 발생하는 경우
보통의공대생
보통의공대생
수학,프로그래밍,기계항공우주 등 공부하는 기록들을 남깁니다.
  • 보통의공대생
    뛰는 놈 위에 나는 공대생
    보통의공대생
  • 전체
    오늘
    어제
    • 분류 전체보기 (468)
      • 공지 (1)
      • 영어 공부 English Study (40)
        • 텝스 TEPS (7)
        • 글 Article (21)
        • 영상 Video (10)
      • 연구 Research (99)
        • 최적화 Optimization (3)
        • 데이터과학 Data Science (7)
        • 인공지능 Artificial Intelligent (40)
        • 제어 Control (45)
      • 프로그래밍 Programming (103)
        • 매트랩 MATLAB (25)
        • 파이썬 Python (33)
        • 줄리아 Julia (2)
        • C++ (3)
        • 리눅스 우분투 Ubuntu (6)
      • 항공우주 Aeronautical engineeri.. (21)
        • 항법 Navigation (0)
        • 유도 Guidance (0)
      • 기계공학 Mechanical engineering (13)
        • 열역학 Thermodynamics (0)
        • 고체역학 Statics & Solid mechan.. (10)
        • 동역학 Dynamics (1)
        • 유체역학 Fluid Dynamics (0)
      • 수학 Mathematics (34)
        • 선형대수학 Linear Algebra (18)
        • 미분방정식 Differential Equation (3)
        • 확률및통계 Probability & Sta.. (2)
        • 미적분학 Calculus (1)
        • 복소해석학 Complex Analysis (5)
        • 실해석학 Real Analysis (0)
      • 수치해석 Numerical Analysis (27)
      • 확률 및 랜덤프로세스 Random process (2)
      • 추론 & 추정 이론 Estimation (3)
      • 기타 (26)
        • 설계 프로젝트 System Design (8)
        • 논문작성 Writing (55)
        • 세미나 Seminar (2)
        • 생산성 Productivity (3)
      • 실험 Experiment (1)
      • 유학 생활 Daily (8)
  • 블로그 메뉴

    • 홈
    • 태그
    • 방명록
  • 링크

  • 공지사항

  • 인기 글

  • 태그

    수치해석
    딥러닝
    텝스공부
    Linear algebra
    JAX
    우분투
    ChatGPT
    에러기록
    Dear abby
    Python
    teps
    논문작성법
    Statics
    Numerical Analysis
    서버
    고체역학
    옵시디언
    MATLAB
    pytorch
    obsidian
    논문작성
    matplotlib
    텝스
    생산성
    Julia
    LaTeX
    Zotero
    인공지능
    IEEE
    WOX
  • 최근 댓글

  • 최근 글

  • hELLO· Designed By정상우.v4.10.3
보통의공대생
[에러기록] assertionerror: if capturable=false, state_steps should not be cuda tensors.
상단으로

티스토리툴바