From 9ba9305370eee4de788e341209f05d060d298a69 Mon Sep 17 00:00:00 2001 From: xusu Date: Tue, 18 Aug 2020 12:24:48 +0800 Subject: [PATCH] Fix lr to 8gpus in slowonly --- .../slowonly/slowonly_r50_4x16x1_256e_kinetics400_rgb.py | 4 ++-- .../slowonly/slowonly_r50_8x8x1_256e_kinetics400_rgb.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/configs/recognition/slowonly/slowonly_r50_4x16x1_256e_kinetics400_rgb.py b/configs/recognition/slowonly/slowonly_r50_4x16x1_256e_kinetics400_rgb.py index ee721d2be2..d171011ae0 100644 --- a/configs/recognition/slowonly/slowonly_r50_4x16x1_256e_kinetics400_rgb.py +++ b/configs/recognition/slowonly/slowonly_r50_4x16x1_256e_kinetics400_rgb.py @@ -71,7 +71,7 @@ dict(type='ToTensor', keys=['imgs']) ] data = dict( - videos_per_gpu=16, + videos_per_gpu=8, workers_per_gpu=4, train=dict( type=dataset_type, @@ -90,7 +90,7 @@ pipeline=test_pipeline)) # optimizer optimizer = dict( - type='SGD', lr=0.6, momentum=0.9, + type='SGD', lr=0.1, momentum=0.9, weight_decay=0.0001) # this lr is used for 8 gpus optimizer_config = dict(grad_clip=dict(max_norm=40, norm_type=2)) # learning policy diff --git a/configs/recognition/slowonly/slowonly_r50_8x8x1_256e_kinetics400_rgb.py b/configs/recognition/slowonly/slowonly_r50_8x8x1_256e_kinetics400_rgb.py index f43f3ae13e..02c50db20d 100644 --- a/configs/recognition/slowonly/slowonly_r50_8x8x1_256e_kinetics400_rgb.py +++ b/configs/recognition/slowonly/slowonly_r50_8x8x1_256e_kinetics400_rgb.py @@ -71,7 +71,7 @@ dict(type='ToTensor', keys=['imgs']) ] data = dict( - videos_per_gpu=16, + videos_per_gpu=8, workers_per_gpu=4, train=dict( type=dataset_type, @@ -90,7 +90,7 @@ pipeline=test_pipeline)) # optimizer optimizer = dict( - type='SGD', lr=0.6, momentum=0.9, + type='SGD', lr=0.1, momentum=0.9, weight_decay=0.0001) # this lr is used for 8 gpus optimizer_config = dict(grad_clip=dict(max_norm=40, norm_type=2)) # learning policy