CalvinBackup/neuralchen-SimSwap
1
0
Fork 0
Code Issues 288 Pull Requests 16 Actions Packages Projects Releases 2 Wiki Activity

Update m1_tf_test.py

Browse Source
This commit is contained in:
Smiril
2022-07-10 22:58:22 +02:00
committed by GitHub
parent 8e8e0e08a2
commit 12eaf87335
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
m1_tf_test.py
+2 -2
View File
@@ -32,8 +32,8 @@ loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
model.compile(optimizer='adam', model.compile(optimizer='adam',
loss=loss_fn, loss=loss_fn,
metrics=['accuracy']) metrics=['accuracy'])
model.fit(x_test, y_test, epochs=10) model.fit(x_train, y_train, epochs=10)
outputs = tf.keras.layers.Dense(4, activation='softmax', name='predictions')((x_test,y_test)) outputs = tf.keras.layers.Dense(4, activation='softmax', name='predictions')((x_train,y_train))
model = tf.keras.Model(inputs=inputs, outputs=outputs) model = tf.keras.Model(inputs=inputs, outputs=outputs)
model.build() model.build()
model.save('model') model.save('model')