Update Model.py

Model.py

@@ -207,7 +207,7 @@ class LoU(layers.Layer):
 
         q = self.Qr(q)
         k = self.Kr(k)
-        v = self.Vr(v)
+        V = self.Vr(V)
 
         # gating signals in (0,1)
         g_q = tf.nn.sigmoid(q)
@@ -328,13 +328,8 @@ dummy_input = np.zeros((1, max_len), dtype=np.int32)
 model(dummy_input)
 model.summary()
 
-# 학습 시작
-history = model.fit(
-    dataset,
-    epochs=1,
-    steps_per_epoch = encoded_inputs.shape[0] // batch_size,
-    verbose=1
-)
+history = model.fit(dataset, epochs=1, verbose=1)
+
 
 # 가중치 저장
 model.save_weights("Cobra.weights.h5")