SYMBOL INDEX (68 symbols across 10 files)

FILE: nn_transfer/test/architectures/lenet.py
  class LeNetPytorch (line 13) | class LeNetPytorch(nn.Module):
    method __init__ (line 14) | def __init__(self):
    method forward (line 22) | def forward(self, x):
  function lenet_keras (line 34) | def lenet_keras():

FILE: nn_transfer/test/architectures/simplenet.py
  class SimpleNetPytorch (line 13) | class SimpleNetPytorch(nn.Module):
    method __init__ (line 14) | def __init__(self):
    method forward (line 20) | def forward(self, x):
  function simplenet_keras (line 28) | def simplenet_keras():

FILE: nn_transfer/test/architectures/unet.py
  function unet_keras (line 16) | def unet_keras(input_size=224):
  class UNetConvBlock (line 92) | class UNetConvBlock(nn.Module):
    method __init__ (line 93) | def __init__(self, in_size, out_size, kernel_size=3, activation=F.relu):
    method forward (line 99) | def forward(self, x):
  class UNetUpBlock (line 105) | class UNetUpBlock(nn.Module):
    method __init__ (line 106) | def __init__(self, in_size, out_size, kernel_size=3,
    method center_crop (line 114) | def center_crop(self, layer, target_size):
    method forward (line 119) | def forward(self, x, bridge):
  class UNetPytorch (line 129) | class UNetPytorch(nn.Module):
    method __init__ (line 130) | def __init__(self):
    method forward (line 154) | def forward(self, x):

FILE: nn_transfer/test/architectures/vggnet.py
  function vggnet_pytorch (line 11) | def vggnet_pytorch():
  function vggnet_keras (line 15) | def vggnet_keras():

FILE: nn_transfer/test/helpers.py
  function set_seeds (line 18) | def set_seeds():
  class TransferTestCase (line 23) | class TransferTestCase(object):
    method assertEqualPrediction (line 24) | def assertEqualPrediction(
    method is_keras_to_pytorch (line 44) | def is_keras_to_pytorch(self):
    method transfer (line 47) | def transfer(self, keras_model, pytorch_model, verbose=False):

FILE: nn_transfer/test/test_architectures.py
  class TestArchitectures (line 12) | class TestArchitectures(TransferTestCase, unittest.TestCase):
    method setUp (line 14) | def setUp(self):
    method test_simplenet (line 20) | def test_simplenet(self):
    method test_lenet (line 31) | def test_lenet(self):
    method test_unet (line 41) | def test_unet(self):
    method test_vggnet (line 52) | def test_vggnet(self):

FILE: nn_transfer/test/test_layers.py
  class BatchNet (line 16) | class BatchNet(nn.Module):
    method __init__ (line 17) | def __init__(self):
    method forward (line 21) | def forward(self, x):
  class ELUNet (line 25) | class ELUNet(nn.Module):
    method __init__ (line 26) | def __init__(self):
    method forward (line 30) | def forward(self, x):
  class TransposeNet (line 34) | class TransposeNet(nn.Module):
    method __init__ (line 35) | def __init__(self):
    method forward (line 39) | def forward(self, x):
  class PReLUNet (line 43) | class PReLUNet(nn.Module):
    method __init__ (line 44) | def __init__(self):
    method forward (line 48) | def forward(self, x):
  class Conv2DNet (line 52) | class Conv2DNet(nn.Module):
    method __init__ (line 53) | def __init__(self):
    method forward (line 57) | def forward(self, x):
  class Conv3DNet (line 61) | class Conv3DNet(nn.Module):
    method __init__ (line 62) | def __init__(self):
    method forward (line 66) | def forward(self, x):
  class TestLayers (line 70) | class TestLayers(TransferTestCase, unittest.TestCase):
    method setUp (line 72) | def setUp(self):
    method test_batch_normalization (line 76) | def test_batch_normalization(self):
    method test_transposed_conv (line 89) | def test_transposed_conv(self):
    method test_elu (line 102) | def test_elu(self):
    method test_prelu (line 114) | def test_prelu(self):
    method test_conv2d (line 126) | def test_conv2d(self):
    method test_conv3d (line 142) | def test_conv3d(self):
    method test_keras_model_changed_as_expected (line 157) | def test_keras_model_changed_as_expected(self):

FILE: nn_transfer/test/test_util.py
  class TestUtil (line 6) | class TestUtil(unittest.TestCase):
    method test_state_dict_names (line 8) | def test_state_dict_names(self):

FILE: nn_transfer/transfer.py
  function check_for_missing_layers (line 24) | def check_for_missing_layers(keras_names, pytorch_layer_names, verbose):
  function keras_to_pytorch (line 36) | def keras_to_pytorch(keras_model, pytorch_model,
  function pytorch_to_keras (line 104) | def pytorch_to_keras(pytorch_model, keras_model,
  function convert_weights (line 171) | def convert_weights(weights, to_keras, flip_filters, flip_channels=False):

FILE: nn_transfer/util.py
  function state_dict_layer_names (line 7) | def state_dict_layer_names(state_dict):
  function _contains_weights (line 13) | def _contains_weights(keras_h5_layer):
  function dig_to_params (line 20) | def dig_to_params(keras_h5_layer):