Commit ca31cd0f authored by pvskand's avatar pvskand
Browse files

adding slow attr to unit tests of unet and resnet

parent add2f183
Loading
Loading
Loading
Loading
+2 −0
Original line number Diff line number Diff line
@@ -2,12 +2,14 @@ import deepchem
import numpy as np
import tensorflow as tf
import unittest
from nose.plugins.attrib import attr
from deepchem.models.tensorgraph.models import resnet50
from deepchem.models.tensorgraph import layers


class TestResNet50(unittest.TestCase):

  @attr('slow')
  def test_resnet50(self):
    resnet = deepchem.models.tensorgraph.models.resnet50.ResNet50(
        learning_rate=0.003,
+2 −0
Original line number Diff line number Diff line
@@ -2,12 +2,14 @@ import deepchem
import numpy as np
import tensorflow as tf
import unittest
from nose.plugins.attrib import attr
from deepchem.models.tensorgraph.models import unet
from deepchem.models.tensorgraph import layers


class TestUNet(unittest.TestCase):

  @attr('slow')
  def test_unet(self):
    unet2D = deepchem.models.tensorgraph.models.unet.UNet(
        learning_rate=0.003, img_rows=32, img_cols=32, model_dir='./unet/')