Alexey/IIS_2023_1
4
3
Fork 1
Code Issues Pull Requests 5 Packages Projects Releases Wiki Activity

arutunyan_dmitry_lab_7 is ready

Browse Source
This commit is contained in:
Arutunyan-Dmitry 2023-11-19 21:17:56 +04:00
parent 2a6f06c08b
commit 7a39edd289
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
arutunyan_dmitry_lab_7/main.py
View File

@ -69,11 +69,9 @@ if __name__ == '__main__':
# Создание распределенной стратегии # Создание распределенной стратегии
strategy = tf.distribute.MultiWorkerMirroredStrategy() strategy = tf.distribute.MultiWorkerMirroredStrategy()
# Распределение модели на устройства # Распределение модели на устройства
with strategy.scope(): with strategy.scope():
parallel_model = model parallel_model = model
# Обучение модели на GPU и CPU # Обучение модели на GPU и CPU
parallel_model.fit(X, y, epochs=100, batch_size=256, callbacks=desired_callbacks) parallel_model.fit(X, y, epochs=100, batch_size=256, callbacks=desired_callbacks)