Browse Source

Merge branch 'dev0.8.0' of github.com:fastnlp/fastNLP into dev0.8.0

tags/v1.0.0alpha
yh_cc 3 years ago
parent
commit
c766ba930e
1 changed files with 4 additions and 2 deletions
  1. +4
    -2
      tests/core/controllers/test_trainer_paddle.py

+ 4
- 2
tests/core/controllers/test_trainer_paddle.py View File

@@ -4,9 +4,11 @@ from dataclasses import dataclass
from fastNLP.core.controllers.trainer import Trainer
from fastNLP.core.metrics.accuracy import Accuracy
from fastNLP.core.callbacks.progress_callback import RichCallback
from fastNLP.envs.imports import _NEED_IMPORT_PADDLE

from paddle.optimizer import Adam
from paddle.io import DataLoader
if _NEED_IMPORT_PADDLE:
from paddle.optimizer import Adam
from paddle.io import DataLoader


from tests.helpers.models.paddle_model import PaddleNormalModel_Classification_1


Loading…
Cancel
Save