text_html_output_4.png text_html_output_6.png text_html_output_2.png text_html_output_5.png text_html_output_1.png application_vnd.jupyter.stderr_output_1.png text_html_output_3.png
from torchsummary import summary

import torch

from torchvision import models
from torchsummary import summary
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
model = RN_BCNN().to(device)
summary(model, (3, 224, 224))