Skip to content

Support latest PyTorch and Torchvision API and 1-channel output#321

Open
senarvi wants to merge 1 commit intoNVIDIA:masterfrom
groke-technologies:fix-one-output-channel
Open

Support latest PyTorch and Torchvision API and 1-channel output#321
senarvi wants to merge 1 commit intoNVIDIA:masterfrom
groke-technologies:fix-one-output-channel

Conversation

@senarvi
Copy link
Copy Markdown

@senarvi senarvi commented May 25, 2023

Fixes a few broken things:

  • Works with the latest PyTorch.
  • Works with the latest Torchvision.
  • Fixed support for one output channel (--output_nc 1).

- Works with the latest PyTorch.
- Works with the latest Torchvision.
- Using one output channel works.
@zhangxiaojuan66
Copy link
Copy Markdown

zhangxiaojuan66 commented May 25, 2023 via email

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants