VGG19

[1409.1556] Very Deep Convolutional Networks for Large-Scale Image Recognitionが原論文らしい。モデルの絵がない・・・。torchvision.models.vgg19 でも引っ張ってきて、モデルを TensorBoard で可視化するとかして眺めるしかないか・・・。
Visualizing Models, Data, and Training with TensorBoard — PyTorch Tutorials 1.5.0 documentationが参考になる。

import torch
import torchvision.models as models
from torch.utils.tensorboard import SummaryWriter
import datetime
import pytz

model = models.vgg19(pretrained=True).to(device)

now = datetime.datetime.now(pytz.timezone('UTC'))
now = now.astimezone(pytz.timezone('Asia/Tokyo'))

writer = SummaryWriter(f'logs/{now.strftime("%Y%m%d-%H%M%S")}')

model.eval()
with torch.no_grad():
    writer.add_graph(model, input_img)

model.train()

writer.close()

こんな感じで良いはず。