remove enum reference from trainer

trainerSaveLoadParams
Thuan Nguyen 8 years ago
parent 6d58d6dc48
commit 10cb9424fd

@ -12,15 +12,13 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from enum import Enum
__all__ = [ __all__ = [
'Event', 'Event',
'Trainer', 'Trainer',
] ]
class Event(Enum): class Event(object):
BEGIN_EPOCH = 0 BEGIN_EPOCH = 0
END_EPOCH = 1 END_EPOCH = 1
BEGIN_STEP = 2 BEGIN_STEP = 2

Loading…
Cancel
Save