arutunyan_dmitry_lab_7 is ready

This commit is contained in:
Arutunyan-Dmitry 2023-11-19 21:17:56 +04:00
parent 2a6f06c08b
commit 7a39edd289

View File

@ -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)