Skip to content

Commit

Permalink
readme update
Browse files Browse the repository at this point in the history
  • Loading branch information
Fangyi-Chen committed Mar 6, 2023
1 parent 2476eae commit 99bcede
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ Our config file lies in [configs/sqr](configs/sqr) folder.
We provide two implementation instances of SQR-adamixer in this repo, one is in [/mmdet/models/roi_heads/adamixer_decoder_Qrecycle.py](/mmdet/models/roi_heads/adamixer_decoder_Qrecycle.py), which might be slower for training but require less GPU memory (and easy to understand the logic). Another is in [/mmdet/models/roi_heads/adamixer_decoder_Qrecycle_optimize.py](/mmdet/models/roi_heads/adamixer_decoder_Qrecycle_optimize.py), which is much faster than the former but has higher requirement on GPU memory.

### SQR-Deformable DETR
We provide the implementation of SQR-deformable DETR in `QRDeformableDetrTransformerDecoder` in [/mmdet/models/utils/transformer.py](/mmdet/models/utils/transformer.py). Note Deformable DETR requires 32 batchsize and we follow this setting.
We provide the implementation of SQR-deformable DETR in `QRDeformableDetrTransformerDecoder` in [/mmdet/models/utils/transformer.py](/mmdet/models/utils/transformer.py). Note Deformable DETR requires 32 as training batchsize and we follow this setting.

__NOTE:__
Please use `mmcv_full==1.3.3` and `pytorch>=1.5.0` for correct reproduction.
Expand Down
4 changes: 2 additions & 2 deletions configs/sqr/deformable_detr_SQR_r50_50e_coco.py
Original file line number Diff line number Diff line change
Expand Up @@ -152,8 +152,8 @@
])
]
data = dict(
samples_per_gpu=2,
workers_per_gpu=2,
samples_per_gpu=8,
workers_per_gpu=8,
train=dict(filter_empty_gt=False, pipeline=train_pipeline),
val=dict(pipeline=test_pipeline),
test=dict(pipeline=test_pipeline))
Expand Down

0 comments on commit 99bcede

Please sign in to comment.