Datawhale 春训营 用AI生成化学分子结构

改进模型后准确率变0.2623了



原本预训练模型训练6代草率了

第二次训练时候提高1迭代次数修改为200次

CUDA_ava=0,1,2,3 python main.py --n_epochs 200 \
    --exp_name edm_competition \
    --n_stability_samples 10000 \
    --diffusion_noise_schedule polynomial_2 \
    --diffusion_noise_precision 1e-5 \
    --diffusion_steps 500 \
    --diffusion_loss_type l2 \
    --batch_size 48 \
    --nf 256 \
    --n_layers 12 \
    --lr 1e-4 \
    --normalize_factors '[1,4,10]' \
    --test_epochs 1 \
    --ema_decay 0.9999 \
    --dataset "competition" \
    --datadir "./data"\
    --test_valid_epoch 1 \
    --resume outputs/edm_competition

除此之外，按照教程，改进了下baseline