8prodigy_full版本介绍

以8张512x1024的图作为训练集[全身图],主要训练参数如下

------------------

network_dim = 32

network_alpha = 16

optimizer_type = 'prodigy'

learning_rate = 1

optimizer_args = ["d0=1e-5", "d_coef=0.4"]

lr_scheduler = 'cosine_with_restarts'

lr_scheduler_num_cycles = 4

lr_warmup_steps = 100

batch_size=4

------------------

目标Loss:0.07-0.1

最终Loss:0.107

------------------

========================== 分割线 ==========================