Skip to content

GitLab

Commit 92a69a9c authored by Gunho Park's avatar Gunho Park
Browse files

use_sync_bn=True

parent 006a91ad
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
official/projects/detr/configs/detr.py
View file @ 92a69a9c
...@@ -90,7 +90,7 @@ def detr_coco() -> cfg.ExperimentConfig: ...@@ -90,7 +90,7 @@ def detr_coco() -> cfg.ExperimentConfig:
model=Detr( model=Detr(
num_classes=81, num_classes=81,
input_size=[1333, 1333, 3], input_size=[1333, 1333, 3],
norm_activation=common.NormActivation(use_sync_bn=False)), norm_activation=common.NormActivation()),
losses=Losses(), losses=Losses(),
train_data=coco.COCODataConfig( train_data=coco.COCODataConfig(
tfds_name='coco/2017', tfds_name='coco/2017',
...@@ -161,7 +161,7 @@ def detr_coco() -> cfg.ExperimentConfig: ...@@ -161,7 +161,7 @@ def detr_coco() -> cfg.ExperimentConfig:
'instances_val2017.json'), 'instances_val2017.json'),
model=Detr( model=Detr(
input_size=[1333, 1333, 3], input_size=[1333, 1333, 3],
norm_activation=common.NormActivation(use_sync_bn=False)), norm_activation=common.NormActivation()),
losses=Losses(), losses=Losses(),
train_data=DataConfig( train_data=DataConfig(
input_path=os.path.join(COCO_INPUT_PATH_BASE, 'train*'), input_path=os.path.join(COCO_INPUT_PATH_BASE, 'train*'),
...@@ -225,7 +225,7 @@ def detr_coco() -> cfg.ExperimentConfig: ...@@ -225,7 +225,7 @@ def detr_coco() -> cfg.ExperimentConfig:
model=Detr( model=Detr(
num_classes=81, num_classes=81,
input_size=[1333, 1333, 3], input_size=[1333, 1333, 3],
norm_activation=common.NormActivation(use_sync_bn=False)), norm_activation=common.NormActivation()),
losses=Losses( losses=Losses(
class_offset=1 class_offset=1
), ),
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment