Skip to content

模型训练haze4k参数问题, #37

@applex250

Description

@applex250

请问一下作者haze4k数据集合如何设置训练参数,
batch size 是16
CUDA_VISIBLE_DEVICES=0 python code/train.py --epochs 300 --iters_per_epoch 1000 --finer_eval_step 290000 --w_loss_L1 1.0 --w_loss_CR 0.1 --start_lr 0.0001 --end_lr 0.000001 --model_name DEA-Net-CR2
这样可以吗

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions