This commit is contained in:
Shital Shah 2019-11-13 03:03:23 -08:00
Родитель f43beef0e3
Коммит e840559b2d
3 изменённых файлов: 4 добавлений и 3 удалений

3
.gitignore поставляемый
Просмотреть файл

@ -450,4 +450,5 @@ ASALocalRun/
# BeatPulse healthcheck temp database # BeatPulse healthcheck temp database
healthchecksdb healthchecksdb
runs/ runs/
abc.png

Просмотреть файл

@ -8,7 +8,7 @@ with open("README.md", "r") as fh:
setuptools.setup( setuptools.setup(
name="tensorwatch", name="tensorwatch",
version="0.8.5", version="0.8.6",
author="Shital Shah", author="Shital Shah",
author_email="shitals@microsoft.com", author_email="shitals@microsoft.com",
description="Interactive Realtime Debugging and Visualization for AI", description="Interactive Realtime Debugging and Visualization for AI",

Просмотреть файл

@ -5,6 +5,6 @@ import tensorwatch as tw
vgg16_model = torchvision.models.vgg16() vgg16_model = torchvision.models.vgg16()
drawing = tw.draw_model(vgg16_model, [1, 3, 224, 224]) drawing = tw.draw_model(vgg16_model, [1, 3, 224, 224])
drawing.save('abc') drawing.save('abc.png')
input("Press any key") input("Press any key")