더북(TheBook)

'--lr-scheduler' 또는 '--early-stopping' 인수를 사용할 경우 오차, 정확도 및 모델의 이름으로 사용할 문자열을 지정합니다.

코드 8-30 오차, 정확도 및 모델의 이름에 대한 문자열

if args['lr_scheduler']:
    print('INFO: Initializing learning rate scheduler')
    lr_scheduler = LRScheduler(optimizer)
    loss_plot_name = 'lrs_loss' ------ 학습률 감소를 적용했을 때의 오차에 대한 문자열
    acc_plot_name = 'lrs_accuracy' ------ 학습률 감소를 적용했을 때의 정확도에 대한 문자열
    model_name = 'lrs_model' ------ 학습률 감소를 적용했을 때의 모델에 대한 문자열
if args['early_stopping']:
    print('INFO: Initializing early stopping')
    early_stopping = EarlyStopping()
    loss_plot_name = 'es_loss' ------ 조기 종료를 적용했을 때의 오차에 대한 문자열
    acc_plot_name = 'es_accuracy' ------ 조기 종료를 적용했을 때의 정확도에 대한 문자열
    model_name = 'es_model' ------ 조기 종료를 적용했을 때의 모델에 대한 문자열
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.