image capiton

MikasaLee · May 9, 2024 · 8085124 · 8085124
1 parent 11da1ac
commit 8085124
Show file tree

Hide file tree

Showing 4 changed files with 382 additions and 7 deletions.
diff --git a/lrr_ocr/lrr_SAR/config/sar/sar_resnet31_parallel-decoder_5e_st_handwritting_chinese_recog.py b/lrr_ocr/lrr_SAR/config/sar/sar_resnet31_parallel-decoder_5e_st_handwritting_chinese_recog.py
@@ -29,7 +29,7 @@
     type='ConcatDataset', datasets=test_list, pipeline=_base_.test_pipeline)
 
 default_hooks = dict(
-    logger=dict(type='LoggerHook', interval=1000), # 原本是100，但是batch_size设小的话那这个就输出的太多了。
+    logger=dict(type='LoggerHook', interval=500), # 原本是100，但是batch_size设小的话那这个就输出的太多了。
     checkpoint=dict(type='CheckpointHook',
                     interval=1,
                     by_epoch=True, 
@@ -40,16 +40,16 @@
 )
 
 train_dataloader = dict(
-    batch_size=16,  # 原本是64，小一点。
-    num_workers=64,
+    batch_size=26,  # 原本是64，小一点。
+    num_workers=32,
     persistent_workers=True,
     pin_memory=True,
     sampler=dict(type='DefaultSampler', shuffle=True),
     dataset=train_dataset)   
 
 val_dataloader = dict(
     batch_size=32,
-    num_workers=32,
+    num_workers=16,  # 改回来16
     persistent_workers=True,
     pin_memory=True,
     drop_last=False,
@@ -99,4 +99,3 @@
 _base_.visualizer.vis_backends = [
     dict(type='LocalVisBackend'),
     dict(type='TensorboardVisBackend'),]
-