diff --git a/src/encoders_decoders.py b/src/encoders_decoders.py index b8e33fe..c90ffcc 100755 --- a/src/encoders_decoders.py +++ b/src/encoders_decoders.py @@ -10,7 +10,7 @@ import warnings from tflearn.layers.core import fully_connected, dropout -from tflearn.layers.conv import conv_1d, avg_pool_1d, highway_conv_1d +from tflearn.layers.conv import conv_1d, avg_pool_1d from tflearn.layers.normalization import batch_normalization from tflearn.layers.core import fully_connected, dropout diff --git a/src/point_net_ae.py b/src/point_net_ae.py index d47a6c9..0c789c6 100755 --- a/src/point_net_ae.py +++ b/src/point_net_ae.py @@ -14,7 +14,7 @@ from . autoencoder import AutoEncoder from . in_out import apply_augmentations -from external.structural_losses import nn_distance, approx_match, match_cost +from .. external.structural_losses import nn_distance, approx_match, match_cost class PointNetAutoEncoder(AutoEncoder):