loss = dict(
MutiTaskSigmoidFocalLoss = dict(
weight = 1.0,
...
)
)
metric_args = dict(pred_key=pred_keys,
gt_key=gt_keys[:5],
...)
collect_keys = ['pose_heatmap_for_action', 'pose_heatmap_for_pose', 'image_feature', 'gt_action_upper', 'gt_action_lower', 'gt_pose', 'gt_hand', 'gt_foot']
data_loader = dict(
type = 'ActionDatasetLoader_mtml',
data_folder = '/media/safemotion/HDD5/pjm_test/action_train_dataset_2023/action_kp_imf_split',
...
)
train = dict(
num_workers = 8,
init_lr = 0.1,
batch_size = 32,
epochs = 100,
optimizer = 'SGD',
optimizer_args = dict(momentum=0.9, nesterov=True, weight_decay=0.0001),
scheduler = 'CosineAnnealingLR',
# scheduler = 'StepLR',
scheduler_args = dict(T_max=30, eta_min=0),
adjust_lr_epoch = [10, 50, 100, 130],
adjust_lr_rate = [0.5, 0.1, 0.1, 0.1],
val_interval = 1,
pretrained = None,#'/media/safemotion/HDD5/pjm_test/action_train_test/9.pth',
save_root = '/media/safemotion/HDD5/pjm_test/action_train_result/action_kp_img',
...
)
test = dict(
model_path = None,
save_root = '',
)
