ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 05:18:50 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #1 from HyamsG/patch-1

Browse Source 
constant paths as consts
This commit is contained in:
Allegro AI 2019-06-11 18:12:48 +03:00 committed by GitHub
commit d4e7eecbac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
examples/pytorch_matplotlib.py
View File

@ -59,7 +59,6 @@ import torchvision.models as models
import copy
from trains import Task
task = Task.init(project_name='examples', task_name='pytorch with matplotlib example', task_type=Task.TaskTypes.testing)
@ -96,6 +95,10 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# with name ``images`` in your current working directory.
# desired size of the output image
STYLE_IMAGE_PATH = "./samples/picasso.jpg"
CONTENT_IMAGE_PATH = "./samples/dancing.jpg"
imsize = 512 if torch.cuda.is_available() else 128 # use small size if no gpu
loader = transforms.Compose([
@ -110,8 +113,8 @@ def image_loader(image_name):
return image.to(device, torch.float)
style_img = image_loader("./samples/picasso.jpg")
content_img = image_loader("./samples/dancing.jpg")
style_img = image_loader(STYLE_IMAGE_PATH)
content_img = image_loader(CONTENT_IMAGE_PATH)
assert style_img.size() == content_img.size(), \
"we need to import style and content images of the same size"