arutunyan_dmitry_lab_7 is ready #177

Merged
Alexey merged 5 commits from arutunyan_dmitry_lab_7 into main 2023-12-07 16:11:23 +04:00
Showing only changes of commit 7a39edd289 - Show all commits

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)