(parser)
| 9 | return value |
| 10 | |
| 11 | def parse_transport_args(parser): |
| 12 | group = parser.add_argument_group("Transport arguments") |
| 13 | group.add_argument("--path-type", type=str, default="Linear", choices=["Linear", "GVP", "VP"]) |
| 14 | group.add_argument("--prediction", type=str, default="velocity", choices=["velocity", "score", "noise"]) |
| 15 | group.add_argument("--loss-weight", type=none_or_str, default=None, choices=[None, "velocity", "likelihood"]) |
| 16 | group.add_argument("--sample-eps", type=float) |
| 17 | group.add_argument("--train-eps", type=float) |
| 18 | |
| 19 | def parse_ode_args(parser): |
| 20 | group = parser.add_argument_group("ODE arguments") |
nothing calls this directly
no outgoing calls
no test coverage detected