Yuchan
commited on
Update Model.py
Browse files
Model.py
CHANGED
|
@@ -280,7 +280,7 @@ history = model.fit(dataset, epochs=1, verbose=1)
|
|
| 280 |
|
| 281 |
|
| 282 |
# 가중치 저장
|
| 283 |
-
model.save_weights("
|
| 284 |
print("모델 가중치 저장 완료!")
|
| 285 |
|
| 286 |
def generate_text_topp(model, prompt, max_len=150, max_gen=150, p=0.9, temperature=0.8, min_len=20):
|
|
|
|
| 280 |
|
| 281 |
|
| 282 |
# 가중치 저장
|
| 283 |
+
model.save_weights("model.weights.h5")
|
| 284 |
print("모델 가중치 저장 완료!")
|
| 285 |
|
| 286 |
def generate_text_topp(model, prompt, max_len=150, max_gen=150, p=0.9, temperature=0.8, min_len=20):
|