Skip to content

GitLab

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

fix bug on exporting

parent ea665155
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 3 deletions
src/sdk/pynni/nni/nas/pytorch/pdarts/trainer.py
View file @ 17ea5f0a
# Copyright (c) Microsoft Corporation. # Copyright (c) Microsoft Corporation.
# Licensed under the MIT license. # Licensed under the MIT license.
import logging import logging
from nni.nas.pytorch.callbacks import LRSchedulerCallback from nni.nas.pytorch.callbacks import LRSchedulerCallback
from nni.nas.pytorch.darts import DartsTrainer from nni.nas.pytorch.darts import DartsTrainer
from nni.nas.pytorch.trainer import BaseTrainer from nni.nas.pytorch.trainer import BaseTrainer
...@@ -66,8 +67,5 @@ class PdartsTrainer(BaseTrainer): ...@@ -66,8 +67,5 @@ class PdartsTrainer(BaseTrainer):
def validate(self): def validate(self):
self.model.validate() self.model.validate()
def export(self):
self.mutator.export()
def checkpoint(self): def checkpoint(self):
raise NotImplementedError("Not implemented yet") 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