olivame e1b029201e 权重改为fp32, 其他部分保持bf16 ...

目前参数--encoder_mode有三种选择:
1. fp32: 全部使用fp32, 适合显存充足的情况
2. autocast: 使用torch.cuda.amp.autocast自动混合精度, 稍微快一些, psnr下降较多
3. bf16_full: 全部使用bf16, 精度较高

2026-01-18 20:24:37 +08:00

..

evaluation

权重改为fp32, 其他部分保持bf16

2026-01-18 20:24:37 +08:00

run_base_model_inference.sh

第一次完整测例跑完

2026-01-18 00:30:10 +08:00

run_real_eval_server.sh

第一次完整测例跑完

2026-01-18 00:30:10 +08:00

run_world_model_interaction.sh

第一次完整测例跑完

2026-01-18 00:30:10 +08:00

train.sh

第一次完整测例跑完

2026-01-18 00:30:10 +08:00

trainer.py

第一次完整测例跑完

2026-01-18 00:30:10 +08:00