흐콤이당
60
2019-09-06 00:21:03
0
105

텐서플로 체크포인트 오류입니다 어떻게 해결할 수 있을까요??(초보입니당 ㅠㅠ)



Restoring from checkpoint failed. This is most likely due to a Variable name or other graph key that is missing from the checkpoint. Please ensure that you have not altered the graph expected based on the checkpoint. Original error:

Key bias_1 not found in checkpoint
	 [[node save_1/RestoreV2 (defined at Programing\web_programing\django\django-vegiter\predict\views.py:20) ]]

코드는 아래와 같습니다. flask에서는 잘 돌아가는데 django에서는 위와 같은 오류가 발생합니다. 디렉터리 구조도 똑같습니다.(폴더명이 다른데 혹시 영향이 있을까요?? 바뀐 디렉토리에서 재 학습을 시켜 모델을 저장시켜 놓았습니다.)

views.py 20라인이면 saver = tf.train.Saver() 이 부분인데 무엇이 문제일까요 ㅠㅠ

from django.shortcuts import render, redirect
import datetime
import tensorflow as tf
import numpy as np


# Create your views here.
def index(request):
    # 플레이스 홀더를 설정합니다.
    X = tf.placeholder(tf.float32, shape=[None, 4])
    Y = tf.placeholder(tf.float32, shape=[None, 1])

    W = tf.Variable(tf.random_normal([4, 1]), name="weight")
    b = tf.Variable(tf.random_normal([1]), name="bias")

    # 가설을 설정합니다.
    hypothesis = tf.matmul(X, W) + b

    # 저장된 모델을 불러오는 객체를 선언합니다.
    saver = tf.train.Saver()
    model = tf.global_variables_initializer()

    # 세션 객체를 생성합니다.
    sess = tf.Session()
    sess.run(model)
    # 저장된 모델을 세션에 적용합니다.
    save_path = "./model/saved.cpkt"
    saver.restore(sess, save_path)

    if request.method == "POST":
        avg_temp = float(request.POST['avg_temp'])
        min_temp = float(request.POST['min_temp'])
        max_temp = float(request.POST['max_temp'])
        rain_fall = float(request.POST['rain_fall'])

        # 배추 가격 변수를 선언합니다.
        price = 0

        # 입력된 파라미터를 배열 형태로 준비합니다.
        data = ((avg_temp, min_temp, max_temp, rain_fall), (0, 0, 0, 0))
        arr = np.array(data, dtype=np.float32)

        # 입력 값을 토대로 예측 값을 찾아냅니다.
        x_data = arr[0:4]
        dict = sess.run(hypothesis, feed_dict={X: x_data})

        # 결과 배추 가격을 저장합니다.
        price = dict[0]
    else:
        price = 0
    return render(request, 'predict/index.html', {'price': price})
0
0
  • 답변 0

  • 로그인을 하시면 답변을 등록할 수 있습니다.