diff --git a/examples/export_ResNet18/resnet18.py b/examples/export_ResNet18/resnet18.py index 948a60fa7d56f4624a7941ab696af9e82102f54d..81e335567194a83794bd4be8c2dcbd8056fa3ebf 100644 --- a/examples/export_ResNet18/resnet18.py +++ b/examples/export_ResNet18/resnet18.py @@ -8,7 +8,6 @@ In order for this file to work properly, you should first download the imagenet """ import random -import aidge_core.utils import numpy as np import os import shutil @@ -35,10 +34,7 @@ from aidge_core.export_utils import remove_optional_inputs, get_node_from_metaop # Torch (Dataset) import torch -import torch.nn.functional as F -from torch import nn -from torchvision import transforms, datasets - +from torchvision import transforms # Arguments import argparse