온라인 채널 제품 판매량 예측 AI 온라인 해커톤

무한실행 오류 관련해서

2023.08.10 23:55 1,347 조회

다름이 아니라, 램 사이즈를 키우고 돌려보려고 하니, 램 부족 오류는 뜨지 않는데,

아래 부분의 코드에서 0/1096이 더이상 올라가지 않고 그대로 계속 실행된 상태로 멈추어버리는 현상이 발생하는데,

이유 아시는 분 있으신가요?


model = BaseModel()

optimizer = torch.optim.Adam(params = model.parameters(), lr = CFG["LEARNING_RATE"])

infer_model = train(model, optimizer, train_loader, val_loader, device)


읽어주셔서 감사합니다.

로그인이 필요합니다
0 / 1000
라뀨
2023.08.11 02:24

cpu로 학습시키는거라면 그냥 오래 걸리는걸겁니다. 3060 기준으로 베이스라인 코드 한 에포크에 10분정도 걸리더군요

sepose
2023.08.11 21:41

넵 한번 해볼게요!