Commit 6f34a7a8 authored by Gaëtan Caillaut's avatar Gaëtan Caillaut
Browse files

ajout des imports nécessaires

parent 985bed5a
......@@ -5,6 +5,7 @@ from torch.utils.tensorboard import SummaryWriter
from data import *
from evaluation import *
if __name__ == "__main__":
import argparse
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment