Contents

모델 학습 시 loss가 Nan으로 나올 때 해결법

모델 학습 시 loss 값이 Nan으로 나올 때 해결 방법

  • if df == pandas.DataFrame() df.isnull().any()로 데이터셋에 NaN이나 inf 값이 들어있는지 확인한다.

  • 다른 optimizer들을 사용해본다. (ex. sgd, adam, nadam)

  • 다른 activation function을 사용해본다.

    • 사용하는 모델마다 적합한 activation function이 있습니다.
    • 즉. 모델에 대하여 공부해야하고, 적합한 activation function을 찾아 학습해야 합니다.
  • learning rate(학습률)을 낮춰본다.

  • 데이터 스케일링(정규화 등)을 한 후 시도 한다.