arutunyan_dmitry_lab_7 is ready
This commit is contained in:
parent
2a6f06c08b
commit
7a39edd289
@ -69,11 +69,9 @@ if __name__ == '__main__':
|
||||
|
||||
# Создание распределенной стратегии
|
||||
strategy = tf.distribute.MultiWorkerMirroredStrategy()
|
||||
|
||||
# Распределение модели на устройства
|
||||
with strategy.scope():
|
||||
parallel_model = model
|
||||
|
||||
# Обучение модели на GPU и CPU
|
||||
parallel_model.fit(X, y, epochs=100, batch_size=256, callbacks=desired_callbacks)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user