Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[Fit API]move to gluon contrib #14635

Merged
merged 1 commit into from
Apr 5, 2019
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,9 @@
import copy
import warnings
from .event_handler import EventHandler, LoggingHandler
from ... import gluon, autograd
from ...context import Context, cpu, gpu, num_gpus
from ...metric import EvalMetric, Loss, Accuracy
from .... import gluon, autograd
from ....context import Context, cpu, gpu, num_gpus
from ....metric import EvalMetric, Loss, Accuracy

__all__ = ['Estimator']

Expand Down
2 changes: 1 addition & 1 deletion tests/nightly/estimator/test_estimator_cnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
import mxnet as mx
from mxnet import gluon, init, nd
from mxnet.gluon import data
from mxnet.contrib.estimator import estimator
from mxnet.gluon.contrib.estimator import estimator
from mxnet.gluon.model_zoo import vision

def load_data_mnist(batch_size, resize=None, num_workers=4):
Expand Down
2 changes: 1 addition & 1 deletion tests/nightly/estimator/test_sentiment_rnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@
from mxnet import nd, gluon
from mxnet.contrib import text
from mxnet.gluon import nn, rnn
from mxnet.contrib.estimator import estimator
from mxnet.gluon.contrib.estimator import estimator


class TextCNN(nn.Block):
Expand Down
2 changes: 1 addition & 1 deletion tests/python/unittest/test_gluon_estimator.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
import mxnet as mx
from mxnet import gluon
from mxnet.gluon import nn
from mxnet.contrib.estimator import Estimator, EventHandler
from mxnet.gluon.contrib.estimator import Estimator, EventHandler
from nose.tools import assert_raises


Expand Down
2 changes: 1 addition & 1 deletion tests/python/unittest/test_gluon_event_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@
import mxnet as mx
from mxnet import nd
from mxnet.gluon import nn, loss
from mxnet.contrib.estimator import estimator, event_handler
from mxnet.gluon.contrib.estimator import estimator, event_handler

def _get_test_network():
net = nn.Sequential()
Expand Down