SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

examples/seq2seq/__init__.py mutates sys.path (#7194)

Browse Source
This commit is contained in:
Stas Bekman
2020-09-20 13:54:42 -07:00
committed by GitHub
parent a4faeceaed
commit 7cbf0f722d
8 changed files with 58 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
examples/seq2seq/test_seq2seq_examples.py
View File

@@ -12,16 +12,15 @@ import pytorch_lightning as pl
import torch
import lightning_base
from convert_pl_checkpoint_to_hf import convert_pl_to_hf
from distillation import distill_main, evaluate_checkpoint
from finetune import SummarizationModule, main
from run_eval import generate_summaries_or_translations, run_generate
from run_eval_search import run_search
from transformers import AutoConfig, AutoModelForSeq2SeqLM
from transformers.hf_api import HfApi
from transformers.testing_utils import CaptureStderr, CaptureStdout, require_multigpu, require_torch_and_cuda, slow
from .convert_pl_checkpoint_to_hf import convert_pl_to_hf
from .distillation import distill_main, evaluate_checkpoint
from .finetune import SummarizationModule, main
from .run_eval import generate_summaries_or_translations, run_generate
from .run_eval_search import run_search
from .utils import label_smoothed_nll_loss, lmap, load_json
from utils import label_smoothed_nll_loss, lmap, load_json
logging.basicConfig(level=logging.DEBUG)