Logic/mamba_diffusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: simplify delta-only flow training

Browse Source 
Remove learned dt prediction and auxiliary losses.

Add repository contributor guidelines.
This commit is contained in:
gameloader
2026-03-10 18:23:17 +08:00
parent 913740266b
commit 01fc1e4eab
4 changed files with 94 additions and 170 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@@ -19,11 +19,6 @@ def build_parser() -> argparse.ArgumentParser:
parser.add_argument("--dt-min", type=float, default=None)
parser.add_argument("--dt-max", type=float, default=None)
parser.add_argument("--lambda-flow", type=float, default=None)
parser.add_argument("--lambda-pos", type=float, default=None)
parser.add_argument("--lambda-dt", type=float, default=None)
parser.add_argument("--use-flow-loss", action=argparse.BooleanOptionalAction, default=None)
parser.add_argument("--use-pos-loss", action=argparse.BooleanOptionalAction, default=None)
parser.add_argument("--use-dt-loss", action=argparse.BooleanOptionalAction, default=None)
parser.add_argument("--num-classes", type=int, default=None)
parser.add_argument("--image-size", type=int, default=None)
parser.add_argument("--channels", type=int, default=None)
@@ -41,7 +36,6 @@ def build_parser() -> argparse.ArgumentParser:
parser.add_argument("--val-every", type=int, default=None)
parser.add_argument("--val-samples-per-class", type=int, default=None)
parser.add_argument("--val-grid-rows", type=int, default=None)
parser.add_argument("--val-max-steps", type=int, default=None)
parser.add_argument("--use-ddp", action=argparse.BooleanOptionalAction, default=None)
return parser