Skip to content

GitLab

"git@developer.sourcefind.cn:gaoqiong/composable_kernel.git" did not exist on "8d63552b4ec1d89e2d8ffd67af6efc8777e8d993"
Unverified Commit 4645e283 authored by Prathik Rao's avatar Prathik Rao Committed by GitHub
Browse files

tensor format ort bug fix (#1557) 



bug fix
Co-authored-by: default avatarPrathik Rao <prathikrao@microsoft.com>
Co-authored-by: default avataranton- <anton@huggingface.co>
parent 58933059
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/unconditional_image_generation/train_unconditional_ort.py
View file @ 4645e283
...@@ -77,7 +77,7 @@ def main(args): ...@@ -77,7 +77,7 @@ def main(args):
), ),
) )
model = ORTModule(model) model = ORTModule(model)
noise_scheduler = DDPMScheduler(num_train_timesteps=1000, tensor_format="pt") noise_scheduler = DDPMScheduler(num_train_timesteps=1000)
optimizer = torch.optim.AdamW( optimizer = torch.optim.AdamW(
model.parameters(), model.parameters(),
lr=args.learning_rate, lr=args.learning_rate,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment