-
Notifications
You must be signed in to change notification settings - Fork 518
Optim-wip: Add Neural Style Transfer tutorial & pre-trained VGG-16 model from Caffe #677
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Optim-wip: Add Neural Style Transfer tutorial & pre-trained VGG-16 model from Caffe #677
Conversation
* Update code to use torch.linalg module when available. * Change InceptionV1 transform_input function input range check to warning.
Some of the style images in the tutorial are currently failing to download as Wikipedia is now requiring we supply a user agent with a download request. We'll need to add a user agent header to the |
Hi @ProGamerGov, learned a lot by going through the tutorial! But do you mind providing me more context? What is the goal and why do you believe Captum needs a neural style transfer tutorial? I can see it demonstrates how to leverage some newly added utility & loss functions to redo a similar PyTorch tutorial. However, why can't we show them in a tutorial with stronger taste of model interpretability? |
Hi @aobo-y, thank you for reviewing this tutorial and it's associated code! In terms of the benefits of including an NST tutorial, I have the following:
|
We can probably skip this PR for now. |
The InceptionV1 portion of the tutorial may fail until the input range changes from #655 are implemented.I updated the transform_input function in this PR to match the changes made in the other PR, so #655 is no longer required for this PR.