Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

add BayesianCNN #9

Merged
merged 2 commits into from
Jan 2, 2024
Merged

add BayesianCNN #9

merged 2 commits into from
Jan 2, 2024

Conversation

GreatV
Copy link
Contributor

@GreatV GreatV commented Dec 17, 2023

pytroch 无法导出

AssertionError: Mutating module attribute W_sigma during export.

from user code:
   File "/home/greatx/repos/PyTorch-BayesianCNN/layers/misc.py", line 18, in forward
    x = module(x)
  File "/home/greatx/repos/PyTorch-BayesianCNN/venv/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/greatx/repos/PyTorch-BayesianCNN/layers/BBB_LRT/BBBConv.py", line 64, in forward
    self.W_sigma = torch.log1p(torch.exp(self.W_rho))

Set TORCH_LOGS="+dynamo" and TORCHDYNAMO_VERBOSE=1 for more information

@2742195759
Copy link
Contributor

Paddle呢?是否可以导出呢?

@GreatV
Copy link
Contributor Author

GreatV commented Dec 18, 2023

Paddle呢?是否可以导出呢?

paddle可以的

Copy link
Contributor

@SigureMo SigureMo left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTMeow

@SigureMo SigureMo merged commit d456392 into PaddleJitLab:develop Jan 2, 2024
@GreatV GreatV deleted the add_BayesianCNN branch January 3, 2024 07:52
@SigureMo SigureMo added the HappyOpenSource 快乐开源活动issue与PR label Jan 4, 2024
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
HappyOpenSource 快乐开源活动issue与PR
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants