# pureddpm

**Repository Path**: meng-fluency/pureddpm

## Basic Information

- **Project Name**: pureddpm
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2026-01-31
- **Last Updated**: 2026-02-01

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 纯ddpm
## 训练命令
python pureddpm.py --early_stop_patience 30 --dropout 0.2 --weight_decay 0.0005 --warmup_steps 500 --seed 101 --mode test_and_valid --batch_size 2    --max_steps 8000   --train_fits "K:\BaiduNetdiskDownload\SDC3a\120_msw_image.fits"   --val_fits "K:\BaiduNetdiskDownload\SDC3a\11to11_msw_image.fits" --output_dir "K:\BaiduNetdiskDownload\SDC3a\pureddpm_model"   --fid_model_path K:\BaiduNetdiskDownload\SDC3a\weights-inception-2015-12-05-6726825d.pth

## 测试命令

python pureddpm.py --mode test --model_path seed101_lr0.0001_bs2_steps8000_20251022_182727_model.pth --val_fits "/data/mlc/SKA-SDC-dataset/dirty_fangzhen_1.fits" --train_log_file /data/mlc/Pure-DDPM/seed101_lr0.0001_bs2_steps8000_20251022_182727_train.csv

# ddpm-with_physical_loss

## 训练命令
python pureddpm_physical_loss.py --early_stop_patience 30 --dropout 0.2 --weight_decay 0.0005 --warmup_steps 500 --seed 101 --mode test_and_valid --batch_size 2    --max_steps 8000   --train_fits "/data/mlc/SKA-SDC-dataset/ZW3.msw_image_without_zero.fits" --psf_fits "/data/mlc/SKA-SDC-dataset/ZW3.msw_psf_without_zero.fits"   --val_fits "/data/mlc/SKA-SDC-dataset/dirty_fangzhen.fits" --output_dir "./pureddpm_physical_loss"   --fid_model_path ./pureddpm_physical_loss/model_weight.pth

## 测试命令
python pureddpm_physical_loss.py --mode test --model_path ./pureddpm_physical_loss/seed101_lr0.0001_bs2_steps8000_20260124_222338_best_model.pth --val_fits "/data/mlc/SKA-SDC-dataset/dirty_fangzhen_1.fits" --train_log_file ./pureddpm_physical_loss

# ddpm-with cross attention

## 训练命令
CUDA_VISIBLE_DEVICES=1,2 python pureddpm_cross_attention.py --early_stop_patience 30 --dropout 0.2 --weight_decay 0.0005 --warmup_steps 500 --seed 101 --mode test_and_valid --batch_size 2    --max_steps 8000   --train_fits "/data/mlc/SKA-SDC-dataset/ZW3.msw_image_without_zero.fits" --psf_fits "/data/mlc/SKA-SDC-dataset/ZW3.msw_psf_without_zero.fits" --use_cross_attention --val_fits "/data/mlc/SKA-SDC-dataset/dirty_fangzhen.fits" --output_dir "./pureddpm_cross_attention"   --fid_model_path ./pureddpm_cross_attention/model_weight.pth

## 测试命令
CUDA_VISIBLE_DEVICES=1,2 python pureddpm_cross_attention.py --mode test --model_path ./pureddpm_cross_attention/seed101_lr0.0001_bs2_steps8000_20260131_213258_best_model.pth --val_fits "/data/mlc/SKA-SDC-dataset/dirty_fangzhen_1.fits" --psf_fits "/data/mlc/SKA-SDC-dataset/psf_fangzhen.fits" --train_log_file ./pureddpm_cross_attention --use_cross_attention