분자구조 이미지 SMILES 변환 AI 경진대회

BASELINE 코드 질문드립니다.

2020.09.14 13:14 4,825 조회

안녕하세요 BASELINE 코드중에서 Epoch돌리는 부분에서 에러가 나는데요... ㅠㅠ


GPU에 사용에 문제가 생긴걸까요...?


gpu :2080 super

cuda : 10.1

cudnn: 7.6.5


로그인이 필요합니다
0 / 1000
bbchip13
2020.09.14 14:46

GPU 메모리 오버플로우 관련 문제로 보이네요 ㅎㅎ
배치 사이즈를 조금 낮춰서 다시 돌려보시면 돌아갈 듯 합니다 

포도당
2020.09.14 15:21

감사합니다 ㅎㅎ

김웅곤
2020.09.14 22:20

batch size, embedding layer를 낮추셔야 합니다.

포도당
2020.09.15 09:19

감사합니다 ㅎㅎ

포도당
2020.09.17 11:50

혹시 배치 몇으로 돌리셨나요?? 저는 64로 하니까 돌아가네요