10
ci/woodpecker/push/woodpecker Pipeline was successful
Details
ci/woodpecker/push/woodpecker Pipeline was successful
Details
This commit is contained in:
parent
67f6ecbbb9
commit
6d896e921b
2
run.py
2
run.py
|
@ -71,7 +71,7 @@ with tf.device('/GPU:0'):
|
|||
model.compile(optimizer='SGD',
|
||||
loss='categorical_crossentropy',
|
||||
metrics=['accuracy'])
|
||||
model.fit(X_train_scaled, y_train_categorical, epochs=50)
|
||||
model.fit(X_train_scaled, y_train_categorical, epochs=10)
|
||||
model.save('mymodel.keras')
|
||||
|
||||
print("finished training")
|
||||
|
|
Loading…
Reference in New Issue