回答編集履歴
1
[Fix] model -> t_model
test
CHANGED
|
@@ -16,7 +16,7 @@
|
|
|
16
16
|
|
|
17
17
|
# data_x, data_y = load_dataset()
|
|
18
18
|
# t_model.fit(data_x, data_y, epoch=10, batch_size=4)
|
|
19
|
-
model.save_weights("weights.hdf5")
|
|
19
|
+
t_model.save_weights("weights.hdf5")
|
|
20
20
|
```
|
|
21
21
|
|
|
22
22
|
## 推論時
|
|
@@ -41,7 +41,7 @@
|
|
|
41
41
|
学習用のモデルで重みをロードすれば推論用のモデルでも重みがロードされる。
|
|
42
42
|
|
|
43
43
|
```python:predict.py
|
|
44
|
-
model.load_weights("weights.hdf5")
|
|
44
|
+
t_model.load_weights("weights.hdf5")
|
|
45
45
|
|
|
46
46
|
is_equal_weight = t_model_base_network.layers[-1].weights[0] == p_model_base_network.layers[-1].weights[0]
|
|
47
47
|
print(is_equal_weight)
|