search

JiaquanYe / TableMASTER-mmocr

2nd solution of ICDAR 2021 Competition on Scientific Literature Parsing, Task B.
Apache License 2.0
410 stars 100 forks source link

请问你是用的什么gpu？我一张2080ti训练表格的那部分只能开到batchsize=4，你这里开到20这么大 #30

Closed cqray1990 closed 2 years ago

cqray1990 commented 2 years ago 
samples_per_gpu=20,
workers_per_gpu=2,
JiaquanYe commented 2 years ago 
samples_per_gpu=20,
workers_per_gpu=2,

In our experience, batchsize can be reach 8 in tesla v100 (16G). It seems make sense that batchsize=4 in 2080ti(11G).

cqray1990 commented 2 years ago

casuse i see your default = 20, so i doubt something wrong with my machine