Параметры запуска

Параметры запуска#

Выбор параметров обучения осуществляется в файле config\config.yaml. Пример файла конфигурации:

### Basic
Mode: 1               # 1 - interactive, 2 - train, 3 - eval
gpu_index: 0

### Train mode
LoadParameters: false
NumTrainRounds: -1    # -1 - infinite
NumRoundEpoch: 15
### Eval mode
NumEvalRounds: 300

### Model parameters
Model: yolo           # yolo
InitWeights: true
LossRatio: 0.04

### Dataset parameters
Dataset: gen1
TimeStep: 16
NumLoadFile: 8
NumWorkers: 4
# Train dataset
BatchSize: 8
NumSteps: 32
TimeWindow: 8
# Test dataset
TestBatchSize: 64
TestNumSteps: 32

### Progress board parameters
Display: True
EveryN: 4

### Plotter parameters
PlotterThreshold: 0.8
PlotterColumns: 4

### Trainer board parameters
EpochSize: 60

Основные параметры

Название

Описание

Mode

Выбор сценария обучения (1 - interactive, 2 - train, 3 - eval)

Dataset

Выбор конфигурации сети (yolo)

LossRatio

Весовой коэффициент для настройки влияния на ошибку ложных срабатываний. Используется для расчёта ошибки следующим образом loss = gt_loss * loss_ratio + background_loss * (1 - self.loss_ratio) + .... Чем больше этот коэффициент, тем меньше влияние ложных срабатываний. В результате сеть делает больше предсказаний, что позволяет повысить активность последнего слоя. Рекомендуется начинать с малых значений и постепенно увеличивать его в ходе обучения.

Dataset

Выбор датасета (gen1, 1mpx)

TimeStep

Временной интервал между кадрами в миллисекундах

BatchSize

Размер пакета для обучения

NumSteps

Количество кадров в одном примере

TimeWindow

Максимальное количество кадров, на которое может уменьшиться последовательность. Это используется для создание обучающих примеров разной длинны.