This commit is contained in:
Nataniel Ruiz
2019-12-25 20:25:08 -04:00
parent 3fa97dc3af
commit 00e6b4a164
+3 -1
View File
@@ -1,7 +1,9 @@
from .base_model import BaseModel
from . import networks
from util import attacks
import numpy as np
import torch
import torch.nn.functional as F
class TestModel(BaseModel):
""" This TesteModel can be used to generate CycleGAN results for only one direction.