Commit 17ea5f0a authored by squirrelsc's avatar squirrelsc
Browse files

fix bug on exporting

parent ea665155
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
import logging
from nni.nas.pytorch.callbacks import LRSchedulerCallback
from nni.nas.pytorch.darts import DartsTrainer
from nni.nas.pytorch.trainer import BaseTrainer
......@@ -66,8 +67,5 @@ class PdartsTrainer(BaseTrainer):
def validate(self):
self.model.validate()
def export(self):
self.mutator.export()
def checkpoint(self):
raise NotImplementedError("Not implemented yet")
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment