Skip to content

Commit

Permalink
pix2pix bug fix
Browse files Browse the repository at this point in the history
  • Loading branch information
dingguanglei committed Dec 28, 2018
1 parent a3c838a commit a229826
Show file tree
Hide file tree
Showing 3 changed files with 38 additions and 0 deletions.
Empty file added unittests/__init__.py
Empty file.
12 changes: 12 additions & 0 deletions unittests/test_featureVisualization.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
from unittest import TestCase


class TestFeatureVisualization(TestCase):
def test__hook(self):
self.fail()

def test__register_forward_hook(self):
self.fail()

def test_trace_activation(self):
self.fail()
26 changes: 26 additions & 0 deletions unittests/test_instences.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
from unittest import TestCase
from jdit.trainer.instances import start_fashingClassTrainer, start_fashingGenerateGanTrainer, \
start_cifarPix2pixGanTrainer
import shutil
import os


class TestInstances(TestCase):
def test_start_fashingClassTrainer(self):
start_fashingClassTrainer(run_type="debug")

def test_start_fashingGenerateGanTrainer(self):
start_fashingGenerateGanTrainer(run_type="debug")

def test_start_cifarPix2pixGanTrainer(self):
start_cifarPix2pixGanTrainer(run_type="debug")

def setUp(self):
dir = "log_debug/"
if os._exists(dir):
shutil.rmtree(dir)

def tearDown(self):
dir = "log_debug/"
if os._exists(dir):
shutil.rmtree(dir)

0 comments on commit a229826

Please sign in to comment.