Code

[OTX] Enable multi-GPU training#1392

Code

Merged

eunwoosh merged 23 commits intofeature/otxopenvinotoolkit/training_extensions:feature/otxfrom multigpu_enablementopenvinotoolkit/training_extensions:multigpu_enablementCopy head branch name to clipboardDec 28, 2022

+372-41

Commits

Commits on Dec 26, 2022

• make logfile saved in save-model-to directory

  eunwoosh

  committed
  867a239

  Copy full SHA for 867a239

• enable train

  eunwoosh

  committed
  94fbfa5

  Copy full SHA for 94fbfa5

• make main process train also

  eunwoosh

  committed
  2ea48e1

  Copy full SHA for 2ea48e1

• bugfix

  eunwoosh

  committed
  d07e116

  Copy full SHA for d07e116

• refactor multi gpu training

  eunwoosh

  committed
  78511fa

  Copy full SHA for 78511fa

• make all processs have same output path

  eunwoosh

  committed
  562621d

  Copy full SHA for 562621d

• prevent child process from being termated by fokred main process

  eunwoosh

  committed
  72805e3

  Copy full SHA for 72805e3

• refactor multigpu implementation

  eunwoosh

  committed
  67fe228

  Copy full SHA for 67fe228

• refactor multi gpu implementation

  eunwoosh

  committed
  c1d913a

  Copy full SHA for c1d913a

• modify argument help sentence

  eunwoosh

  committed
  6b26786

  Copy full SHA for 6b26786

• add multi gpu test code

  eunwoosh

  committed
  fb4bac5

  Copy full SHA for fb4bac5

• align with pre-commit test

  eunwoosh

  committed
  35e0814

  Copy full SHA for 35e0814

• separate multi GPU manager class

  eunwoosh

  committed
  acd54b9

  Copy full SHA for acd54b9

• modify train cli argument 'save-logs-to' to 'output-path'

  eunwoosh

  committed
  ce7acf6

  Copy full SHA for ce7acf6

• remove tray excpet during killing child process

  eunwoosh

  committed
  3e4b6e2

  Copy full SHA for 3e4b6e2

• apply output_path to all tasks

  eunwoosh

  committed
  5ca59dd

  Copy full SHA for 5ca59dd

• change print to logger

  eunwoosh

  committed
  6d1445a

  Copy full SHA for 6d1445a

• skip multi gpu test if number of gpu is insufficient

  eunwoosh

  committed
  497ae13

  Copy full SHA for 497ae13

• fix typo

  eunwoosh

  committed
  428c524

  Copy full SHA for 428c524

• multi gpu test bugfix

  eunwoosh

  committed
  5587d32

  Copy full SHA for 5587d32

• isort fix

  eunwoosh

  committed
  a202a30

  Copy full SHA for a202a30

• test case bugfix

  eunwoosh

  committed
  58b4028

  Copy full SHA for 58b4028

Commits on Dec 28, 2022

• fix typo and change some variable name

  eunwoosh

  committed
  a4fa37c

  Copy full SHA for a4fa37c