search

cuiziteng / Illumination-Adaptive-Transformer

[BMVC 2022] You Only Need 90K Parameters to Adapt Light: A Light Weight Transformer for Image Enhancement and Exposure Correction. SOTA for low light enhancement, 0.004 seconds try this for pre-processing.
Apache License 2.0
441 stars 43 forks source link

training #38

Closed PeterSmith1 closed 8 months ago

PeterSmith1 commented 1 year ago

大佬好，我看你训练IAT-YOLOV3是单卡训练，我拿单卡3090去训练，时间太长了，如果要八卡一起训练的话，我采用你训练detr的训练参数文件，就会出错，如果想八卡同时训练，该怎么设置呢

cuiziteng commented 1 year ago

您好，IAT-YOLOV3单卡的话一般四五个小时就好，八卡detr您那里报错是什么，可以看一下吗

PeterSmith1 commented 1 year ago

image 我是直接拿detr的多卡训练文件去训练YOLO的

cuiziteng commented 1 year ago

看样子是有一部分的参数没有进入梯度传播，你可以先把find_unused_parameter=True这部分设置成False