Skip to content
Snippets Groups Projects
Commit 626a07fe authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Clean import in resnet18 script.

parent de293155
No related branches found
No related tags found
2 merge requests!490.3.1,!32[Feat] Add quantized exports for Resnet18 and LeNet
Pipeline #72768 failed
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment