
'''simple docstring'''





import argparse
import os

import gluonnlp as nlp
import mxnet as mx
import numpy as np
import torch
from gluonnlp.base import get_home_dir
from gluonnlp.model.bert import BERTEncoder
from gluonnlp.model.utils import _load_vocab
from gluonnlp.vocab import Vocab
from packaging import version
from torch import nn

from transformers import BertConfig, BertForMaskedLM, BertModel, RobertaTokenizer
from transformers.models.bert.modeling_bert import (
    BertIntermediate,
    BertLayer,
    BertOutput,
    BertSelfAttention,
    BertSelfOutput,
)
from transformers.utils import logging


if version.parse(nlp.__version__) != version.parse("0.8.3"):
						raise Exception("requires gluonnlp == 0.8.3")

if version.parse(mx.__version__) != version.parse("1.5.0"):
						raise Exception("requires mxnet == 1.5.0")

logging.set_verbosity_info()
UpperCAmelCase    					=   logging.get_logger(__name__)

UpperCAmelCase    					=   "The Nymphenburg Palace is a beautiful palace in Munich!"

def      A (       A_							:					str    ,							A_							:					str       ):
						snake_case       :     str =    {
						    '''attention_cell''': '''multi_head''',
						    '''num_layers''': 4,
						    '''units''': 1024,
						    '''hidden_size''': 768,
						    '''max_length''': 512,
						    '''num_heads''': 8,
						    '''scaled''': True,
						    '''dropout''': 0.1,
						    '''use_residual''': True,
						    '''embed_size''': 1024,
						    '''embed_dropout''': 0.1,
						    '''word_embed''': None,
						    '''layer_norm_eps''': 1e-5,
						    '''token_type_vocab_size''': 2,
						}

						snake_case       :     int =    bort_4_8_768_1024_hparams

						# Let's construct the original Bort model here
						# Taken from official BERT implementation, see:
						# https://github.com/alexa/bort/blob/master/bort/bort.py
						snake_case       :     int =    BERTEncoder(
						    attention_cell=predefined_args['''attention_cell''']    ,							num_layers=predefined_args['''num_layers''']    ,							units=predefined_args['''units''']    ,							hidden_size=predefined_args['''hidden_size''']    ,							max_length=predefined_args['''max_length''']    ,							num_heads=predefined_args['''num_heads''']    ,							scaled=predefined_args['''scaled''']    ,							dropout=predefined_args['''dropout''']    ,							output_attention=A_    ,							output_all_encodings=A_    ,							use_residual=predefined_args['''use_residual''']    ,							activation=predefined_args.get('''activation'''    ,							'''gelu'''       )    ,							layer_norm_eps=predefined_args.get('''layer_norm_eps'''    ,							A_       )    ,							)

						# Vocab information needs to be fetched first
						# It's the same as RoBERTa, so RobertaTokenizer can be used later
						snake_case       :     List[Any] =    '''openwebtext_ccnews_stories_books_cased'''

						# Specify download folder to Gluonnlp's vocab
						snake_case       :     Optional[Any] =    os.path.join(get_home_dir()    ,							'''models'''       )
						snake_case       :     Dict =    _load_vocab(A_    ,							A_    ,							A_    ,							cls=A_       )

						snake_case       :     Tuple =    nlp.model.BERTModel(
						    A_    ,							len(A_       )    ,							units=predefined_args['''units''']    ,							embed_size=predefined_args['''embed_size''']    ,							embed_dropout=predefined_args['''embed_dropout''']    ,							word_embed=predefined_args['''word_embed''']    ,							use_pooler=A_    ,							use_token_type_embed=A_    ,							token_type_vocab_size=predefined_args['''token_type_vocab_size''']    ,							use_classifier=A_    ,							use_decoder=A_    ,							)

						original_bort.load_parameters(A_    ,							cast_dtype=A_    ,							ignore_extra=A_       )
						snake_case       :     Union[str, Any] =    original_bort._collect_params_with_prefix()

						# Build our config 🤗
						snake_case       :     Optional[int] =    {
						    '''architectures''': ['''BertForMaskedLM'''],
						    '''attention_probs_dropout_prob''': predefined_args['''dropout'''],
						    '''hidden_act''': '''gelu''',
						    '''hidden_dropout_prob''': predefined_args['''dropout'''],
						    '''hidden_size''': predefined_args['''embed_size'''],
						    '''initializer_range''': 0.02,
						    '''intermediate_size''': predefined_args['''hidden_size'''],
						    '''layer_norm_eps''': predefined_args['''layer_norm_eps'''],
						    '''max_position_embeddings''': predefined_args['''max_length'''],
						    '''model_type''': '''bort''',
						    '''num_attention_heads''': predefined_args['''num_heads'''],
						    '''num_hidden_layers''': predefined_args['''num_layers'''],
						    '''pad_token_id''': 1,  # 2 = BERT, 1 = RoBERTa
						    '''type_vocab_size''': 1,  # 2 = BERT, 1 = RoBERTa
						    '''vocab_size''': len(A_       ),
						}

						snake_case       :     Optional[int] =    BertConfig.from_dict(A_       )
						snake_case       :     int =    BertForMaskedLM(A_       )
						hf_bort_model.eval()

						# Parameter mapping table (Gluonnlp to Transformers)
						# * denotes layer index
						#
						# | Gluon Parameter                                                | Transformers Parameter
						# | -------------------------------------------------------------- | ----------------------
						# | `encoder.layer_norm.beta`                                      | `bert.embeddings.LayerNorm.bias`
						# | `encoder.layer_norm.gamma`                                     | `bert.embeddings.LayerNorm.weight`
						# | `encoder.position_weight`                                      | `bert.embeddings.position_embeddings.weight`
						# | `word_embed.0.weight`                                          | `bert.embeddings.word_embeddings.weight`
						# | `encoder.transformer_cells.*.attention_cell.proj_key.bias`     | `bert.encoder.layer.*.attention.self.key.bias`
						# | `encoder.transformer_cells.*.attention_cell.proj_key.weight`   | `bert.encoder.layer.*.attention.self.key.weight`
						# | `encoder.transformer_cells.*.attention_cell.proj_query.bias`   | `bert.encoder.layer.*.attention.self.query.bias`
						# | `encoder.transformer_cells.*.attention_cell.proj_query.weight` | `bert.encoder.layer.*.attention.self.query.weight`
						# | `encoder.transformer_cells.*.attention_cell.proj_value.bias`   | `bert.encoder.layer.*.attention.self.value.bias`
						# | `encoder.transformer_cells.*.attention_cell.proj_value.weight` | `bert.encoder.layer.*.attention.self.value.weight`
						# | `encoder.transformer_cells.*.ffn.ffn_2.bias`                   | `bert.encoder.layer.*.attention.output.dense.bias`
						# | `encoder.transformer_cells.*.ffn.ffn_2.weight`                 | `bert.encoder.layer.*.attention.output.dense.weight`
						# | `encoder.transformer_cells.*.layer_norm.beta`                  | `bert.encoder.layer.*.attention.output.LayerNorm.bias`
						# | `encoder.transformer_cells.*.layer_norm.gamma`                 | `bert.encoder.layer.*.attention.output.LayerNorm.weight`
						# | `encoder.transformer_cells.*.ffn.ffn_1.bias`                   | `bert.encoder.layer.*.intermediate.dense.bias`
						# | `encoder.transformer_cells.*.ffn.ffn_1.weight`                 | `bert.encoder.layer.*.intermediate.dense.weight`
						# | `encoder.transformer_cells.*.ffn.layer_norm.beta`              | `bert.encoder.layer.*.output.LayerNorm.bias`
						# | `encoder.transformer_cells.*.ffn.layer_norm.gamma`             | `bert.encoder.layer.*.output.LayerNorm.weight`
						# | `encoder.transformer_cells.*.proj.bias`                        | `bert.encoder.layer.*.output.dense.bias`
						# | `encoder.transformer_cells.*.proj.weight`                      | `bert.encoder.layer.*.output.dense.weight`

						# Helper function to convert MXNET Arrays to PyTorch
						def to_torch(A_							:					List[str]       ) -> nn.Parameter:
												return nn.Parameter(torch.FloatTensor(mx_array.data().asnumpy()       )       )

						# Check param shapes and map new HF param back
						def check_and_map_params(A_							:					Tuple    ,							A_							:					Dict       ):
												snake_case       :     Tuple =    hf_param.shape

												snake_case       :     List[Any] =    to_torch(params[gluon_param]       )
												snake_case       :     int =    gluon_param.shape

												assert (
												    shape_hf == shape_gluon
												), F"""The gluon parameter {gluon_param} has shape {shape_gluon}, but expects shape {shape_hf} for Transformers"""

												return gluon_param

						snake_case       :     List[str] =    check_and_map_params(
						    hf_bort_model.bert.embeddings.word_embeddings.weight    ,							'''word_embed.0.weight'''       )
						snake_case       :     Optional[Any] =    check_and_map_params(
						    hf_bort_model.bert.embeddings.position_embeddings.weight    ,							'''encoder.position_weight'''       )
						snake_case       :     Dict =    check_and_map_params(
						    hf_bort_model.bert.embeddings.LayerNorm.bias    ,							'''encoder.layer_norm.beta'''       )
						snake_case       :     Tuple =    check_and_map_params(
						    hf_bort_model.bert.embeddings.LayerNorm.weight    ,							'''encoder.layer_norm.gamma'''       )

						# Inspired by RoBERTa conversion script, we just zero them out (Bort does not use them)
						snake_case       :     str =    torch.zeros_like(
						    hf_bort_model.bert.embeddings.token_type_embeddings.weight.data       )

						for i in range(hf_bort_config.num_hidden_layers       ):
												snake_case       :     BertLayer =    hf_bort_model.bert.encoder.layer[i]

												# self attention
												snake_case       :     BertSelfAttention =    layer.attention.self

												snake_case       :     Optional[int] =    check_and_map_params(
												    self_attn.key.bias.data    ,							F"""encoder.transformer_cells.{i}.attention_cell.proj_key.bias"""       )

												snake_case       :     Any =    check_and_map_params(
												    self_attn.key.weight.data    ,							F"""encoder.transformer_cells.{i}.attention_cell.proj_key.weight"""       )
												snake_case       :     Tuple =    check_and_map_params(
												    self_attn.query.bias.data    ,							F"""encoder.transformer_cells.{i}.attention_cell.proj_query.bias"""       )
												snake_case       :     Dict =    check_and_map_params(
												    self_attn.query.weight.data    ,							F"""encoder.transformer_cells.{i}.attention_cell.proj_query.weight"""       )
												snake_case       :     int =    check_and_map_params(
												    self_attn.value.bias.data    ,							F"""encoder.transformer_cells.{i}.attention_cell.proj_value.bias"""       )
												snake_case       :     int =    check_and_map_params(
												    self_attn.value.weight.data    ,							F"""encoder.transformer_cells.{i}.attention_cell.proj_value.weight"""       )

												# self attention output
												snake_case       :     BertSelfOutput =    layer.attention.output

												snake_case       :     Optional[Any] =    check_and_map_params(
												    self_output.dense.bias    ,							F"""encoder.transformer_cells.{i}.proj.bias"""       )
												snake_case       :     Tuple =    check_and_map_params(
												    self_output.dense.weight    ,							F"""encoder.transformer_cells.{i}.proj.weight"""       )
												snake_case       :     int =    check_and_map_params(
												    self_output.LayerNorm.bias    ,							F"""encoder.transformer_cells.{i}.layer_norm.beta"""       )
												snake_case       :     Optional[int] =    check_and_map_params(
												    self_output.LayerNorm.weight    ,							F"""encoder.transformer_cells.{i}.layer_norm.gamma"""       )

												# intermediate
												snake_case       :     BertIntermediate =    layer.intermediate

												snake_case       :     Optional[Any] =    check_and_map_params(
												    intermediate.dense.bias    ,							F"""encoder.transformer_cells.{i}.ffn.ffn_1.bias"""       )
												snake_case       :     Optional[int] =    check_and_map_params(
												    intermediate.dense.weight    ,							F"""encoder.transformer_cells.{i}.ffn.ffn_1.weight"""       )

												# output
												snake_case       :     BertOutput =    layer.output

												snake_case       :     List[Any] =    check_and_map_params(
												    bert_output.dense.bias    ,							F"""encoder.transformer_cells.{i}.ffn.ffn_2.bias"""       )
												snake_case       :     int =    check_and_map_params(
												    bert_output.dense.weight    ,							F"""encoder.transformer_cells.{i}.ffn.ffn_2.weight"""       )
												snake_case       :     List[str] =    check_and_map_params(
												    bert_output.LayerNorm.bias    ,							F"""encoder.transformer_cells.{i}.ffn.layer_norm.beta"""       )
												snake_case       :     List[str] =    check_and_map_params(
												    bert_output.LayerNorm.weight    ,							F"""encoder.transformer_cells.{i}.ffn.layer_norm.gamma"""       )

						# Save space and energy 🎄
						hf_bort_model.half()

						# Compare output of both models
						snake_case       :     Dict =    RobertaTokenizer.from_pretrained('''roberta-base'''       )

						snake_case       :     Any =    tokenizer.encode_plus(A_       )['''input_ids''']

						# Get gluon output
						snake_case       :     Dict =    mx.nd.array([input_ids]       )
						snake_case       :     Optional[int] =    original_bort(inputs=A_    ,							token_types=[]       )

						# Get Transformer output (save and reload model again)
						hf_bort_model.save_pretrained(A_       )
						snake_case       :     Optional[int] =    BertModel.from_pretrained(A_       )
						hf_bort_model.eval()

						snake_case       :     Any =    tokenizer.encode_plus(A_    ,							return_tensors='''pt'''       )
						snake_case       :     Any =    hf_bort_model(**A_       )[0]

						snake_case       :     Optional[Any] =    output_gluon[0].asnumpy()
						snake_case       :     Union[str, Any] =    output_hf[0].detach().numpy()

						snake_case       :     Tuple =    np.max(np.abs(hf_layer - gluon_layer       )       ).item()
						snake_case       :     Tuple =    np.allclose(A_    ,							A_    ,							atol=1e-3       )

						if success:
												print('''✔️ Both model do output the same tensors'''       )
						else:
												print('''❌ Both model do **NOT** output the same tensors'''       )
												print('''Absolute difference is:'''    ,							A_       )


if __name__ == "__main__":
						UpperCAmelCase    					=   argparse.ArgumentParser()
						# Required parameters
						parser.add_argument(
						    "--bort_checkpoint_path", default=None, type=str, required=True, help="Path the official Bort params file."
						)
						parser.add_argument(
						    "--pytorch_dump_folder_path", default=None, type=str, required=True, help="Path to the output PyTorch model."
						)
						UpperCAmelCase    					=   parser.parse_args()
						convert_bort_checkpoint_to_pytorch(args.bort_checkpoint_path, args.pytorch_dump_folder_path)



'''simple docstring'''





import argparse
import json
import logging
import os
import shutil
import sys
import tempfile
import unittest
from unittest import mock

import torch
from accelerate.utils import write_basic_config

from transformers.testing_utils import TestCasePlus, get_gpu_count, run_command, slow, torch_device
from transformers.utils import is_apex_available


logging.basicConfig(level=logging.DEBUG)

UpperCAmelCase    					=   logging.getLogger()

def      A (       ):
						snake_case       :     Dict =    argparse.ArgumentParser()
						parser.add_argument('''-f'''       )
						snake_case       :     Any =    parser.parse_args()
						return args.f

def      A (       A_							:					str       ):
						snake_case       :     str =    {}
						snake_case       :     Union[str, Any] =    os.path.join(A_    ,							'''all_results.json'''       )
						if os.path.exists(A_       ):
												with open(A_    ,							'''r'''       ) as f:
																		snake_case       :     List[str] =    json.load(A_       )
						else:
												raise ValueError(F"""can't find {path}"""       )
						return results

def      A (       ):
						snake_case       :     Any =    torch.cuda.is_available() and torch_device == '''cuda'''
						return is_using_cuda and is_apex_available()


UpperCAmelCase    					=   logging.StreamHandler(sys.stdout)
logger.addHandler(stream_handler)





class 						a   (				__magic_name__    ):




				@classmethod
				def 	__snake_case							(							cls	:				Tuple      ):
										# Write Accelerate config, will pick up on CPU, GPU, and multi-GPU
										snake_case       :     int =    tempfile.mkdtemp()
										snake_case       :     Optional[Any] =    os.path.join(cls.tmpdir,    '''default_config.yml'''      )
										write_basic_config(save_location=cls.configPath      )
										snake_case       :     Tuple =    ['''accelerate''', '''launch''', '''--config_file''', cls.configPath]




				@classmethod
				def 	__snake_case							(							cls	:				Optional[int]      ):
										shutil.rmtree(cls.tmpdir      )




				@mock.patch.dict(os.environ,    {'''WANDB_MODE''': '''offline'''}      )
				def 	__snake_case							(							self	:				Any      ):
										snake_case       :     Any =    self.get_auto_remove_tmp_dir()
										snake_case       :     Optional[Any] =    F"""
            {self.examples_dir}/pytorch/text-classification/run_glue_no_trainer.py
            --model_name_or_path distilbert-base-uncased
            --output_dir {tmp_dir}
            --train_file ./tests/fixtures/tests_samples/MRPC/train.csv
            --validation_file ./tests/fixtures/tests_samples/MRPC/dev.csv
            --per_device_train_batch_size=2
            --per_device_eval_batch_size=1
            --learning_rate=1e-4
            --seed=42
            --checkpointing_steps epoch
            --with_tracking
        """.split()

										if is_cuda_and_apex_available():
																testargs.append('''--fp16'''      )

										run_command(self._launch_args + testargs      )
										snake_case       :     int =    get_results(SCREAMING_SNAKE_CASE_      )
										self.assertGreaterEqual(result['''eval_accuracy'''],    0.75      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''epoch_0'''      )      )      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''glue_no_trainer'''      )      )      )




				@mock.patch.dict(os.environ,    {'''WANDB_MODE''': '''offline'''}      )
				def 	__snake_case							(							self	:				Tuple      ):
										snake_case       :     int =    self.get_auto_remove_tmp_dir()
										snake_case       :     str =    F"""
            {self.examples_dir}/pytorch/language-modeling/run_clm_no_trainer.py
            --model_name_or_path distilgpt2
            --train_file ./tests/fixtures/sample_text.txt
            --validation_file ./tests/fixtures/sample_text.txt
            --block_size 128
            --per_device_train_batch_size 5
            --per_device_eval_batch_size 5
            --num_train_epochs 2
            --output_dir {tmp_dir}
            --checkpointing_steps epoch
            --with_tracking
        """.split()

										if torch.cuda.device_count() > 1:
																# Skipping because there are not enough batches to train the model + would need a drop_last to work.
																return

										run_command(self._launch_args + testargs      )
										snake_case       :     List[str] =    get_results(SCREAMING_SNAKE_CASE_      )
										self.assertLess(result['''perplexity'''],    1_00      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''epoch_0'''      )      )      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''clm_no_trainer'''      )      )      )




				@mock.patch.dict(os.environ,    {'''WANDB_MODE''': '''offline'''}      )
				def 	__snake_case							(							self	:				List[Any]      ):
										snake_case       :     Union[str, Any] =    self.get_auto_remove_tmp_dir()
										snake_case       :     int =    F"""
            {self.examples_dir}/pytorch/language-modeling/run_mlm_no_trainer.py
            --model_name_or_path distilroberta-base
            --train_file ./tests/fixtures/sample_text.txt
            --validation_file ./tests/fixtures/sample_text.txt
            --output_dir {tmp_dir}
            --num_train_epochs=1
            --checkpointing_steps epoch
            --with_tracking
        """.split()

										run_command(self._launch_args + testargs      )
										snake_case       :     List[Any] =    get_results(SCREAMING_SNAKE_CASE_      )
										self.assertLess(result['''perplexity'''],    42      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''epoch_0'''      )      )      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''mlm_no_trainer'''      )      )      )




				@mock.patch.dict(os.environ,    {'''WANDB_MODE''': '''offline'''}      )
				def 	__snake_case							(							self	:				Dict      ):
										# with so little data distributed training needs more epochs to get the score on par with 0/1 gpu
										snake_case       :     Any =    7 if get_gpu_count() > 1 else 2

										snake_case       :     Optional[int] =    self.get_auto_remove_tmp_dir()
										snake_case       :     str =    F"""
            {self.examples_dir}/pytorch/token-classification/run_ner_no_trainer.py
            --model_name_or_path bert-base-uncased
            --train_file tests/fixtures/tests_samples/conll/sample.json
            --validation_file tests/fixtures/tests_samples/conll/sample.json
            --output_dir {tmp_dir}
            --learning_rate=2e-4
            --per_device_train_batch_size=2
            --per_device_eval_batch_size=2
            --num_train_epochs={epochs}
            --seed 7
            --checkpointing_steps epoch
            --with_tracking
        """.split()

										run_command(self._launch_args + testargs      )
										snake_case       :     Optional[int] =    get_results(SCREAMING_SNAKE_CASE_      )
										self.assertGreaterEqual(result['''eval_accuracy'''],    0.75      )
										self.assertLess(result['''train_loss'''],    0.5      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''epoch_0'''      )      )      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''ner_no_trainer'''      )      )      )




				@unittest.skip(reason='''Fix me @muellerzr'''      )
				@mock.patch.dict(os.environ,    {'''WANDB_MODE''': '''offline'''}      )
				def 	__snake_case							(							self	:				Tuple      ):
										snake_case       :     Tuple =    self.get_auto_remove_tmp_dir()
										snake_case       :     Optional[int] =    F"""
            {self.examples_dir}/pytorch/question-answering/run_qa_no_trainer.py
            --model_name_or_path bert-base-uncased
            --version_2_with_negative
            --train_file tests/fixtures/tests_samples/SQUAD/sample.json
            --validation_file tests/fixtures/tests_samples/SQUAD/sample.json
            --output_dir {tmp_dir}
            --seed=42
            --max_train_steps=10
            --num_warmup_steps=2
            --learning_rate=2e-4
            --per_device_train_batch_size=2
            --per_device_eval_batch_size=1
            --checkpointing_steps epoch
            --with_tracking
        """.split()

										run_command(self._launch_args + testargs      )
										snake_case       :     str =    get_results(SCREAMING_SNAKE_CASE_      )
										# Because we use --version_2_with_negative the testing script uses SQuAD v2 metrics.
										self.assertGreaterEqual(result['''eval_f1'''],    28      )
										self.assertGreaterEqual(result['''eval_exact'''],    28      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''epoch_0'''      )      )      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''qa_no_trainer'''      )      )      )




				@mock.patch.dict(os.environ,    {'''WANDB_MODE''': '''offline'''}      )
				def 	__snake_case							(							self	:				Optional[int]      ):
										snake_case       :     Any =    self.get_auto_remove_tmp_dir()
										snake_case       :     List[Any] =    F"""
            {self.examples_dir}/pytorch/multiple-choice/run_swag_no_trainer.py
            --model_name_or_path bert-base-uncased
            --train_file tests/fixtures/tests_samples/swag/sample.json
            --validation_file tests/fixtures/tests_samples/swag/sample.json
            --output_dir {tmp_dir}
            --max_train_steps=20
            --num_warmup_steps=2
            --learning_rate=2e-4
            --per_device_train_batch_size=2
            --per_device_eval_batch_size=1
            --with_tracking
        """.split()

										run_command(self._launch_args + testargs      )
										snake_case       :     Union[str, Any] =    get_results(SCREAMING_SNAKE_CASE_      )
										self.assertGreaterEqual(result['''eval_accuracy'''],    0.8      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''swag_no_trainer'''      )      )      )




				@slow
				@mock.patch.dict(os.environ,    {'''WANDB_MODE''': '''offline'''}      )
				def 	__snake_case							(							self	:				str      ):
										snake_case       :     str =    self.get_auto_remove_tmp_dir()
										snake_case       :     Any =    F"""
            {self.examples_dir}/pytorch/summarization/run_summarization_no_trainer.py
            --model_name_or_path t5-small
            --train_file tests/fixtures/tests_samples/xsum/sample.json
            --validation_file tests/fixtures/tests_samples/xsum/sample.json
            --output_dir {tmp_dir}
            --max_train_steps=50
            --num_warmup_steps=8
            --learning_rate=2e-4
            --per_device_train_batch_size=2
            --per_device_eval_batch_size=1
            --checkpointing_steps epoch
            --with_tracking
        """.split()

										run_command(self._launch_args + testargs      )
										snake_case       :     str =    get_results(SCREAMING_SNAKE_CASE_      )
										self.assertGreaterEqual(result['''eval_rouge1'''],    10      )
										self.assertGreaterEqual(result['''eval_rouge2'''],    2      )
										self.assertGreaterEqual(result['''eval_rougeL'''],    7      )
										self.assertGreaterEqual(result['''eval_rougeLsum'''],    7      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''epoch_0'''      )      )      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''summarization_no_trainer'''      )      )      )




				@slow
				@mock.patch.dict(os.environ,    {'''WANDB_MODE''': '''offline'''}      )
				def 	__snake_case							(							self	:				int      ):
										snake_case       :     Any =    self.get_auto_remove_tmp_dir()
										snake_case       :     Optional[int] =    F"""
            {self.examples_dir}/pytorch/translation/run_translation_no_trainer.py
            --model_name_or_path sshleifer/student_marian_en_ro_6_1
            --source_lang en
            --target_lang ro
            --train_file tests/fixtures/tests_samples/wmt16/sample.json
            --validation_file tests/fixtures/tests_samples/wmt16/sample.json
            --output_dir {tmp_dir}
            --max_train_steps=50
            --num_warmup_steps=8
            --num_beams=6
            --learning_rate=3e-3
            --per_device_train_batch_size=2
            --per_device_eval_batch_size=1
            --source_lang en_XX
            --target_lang ro_RO
            --checkpointing_steps epoch
            --with_tracking
        """.split()

										run_command(self._launch_args + testargs      )
										snake_case       :     Optional[int] =    get_results(SCREAMING_SNAKE_CASE_      )
										self.assertGreaterEqual(result['''eval_bleu'''],    30      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''epoch_0'''      )      )      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''translation_no_trainer'''      )      )      )




				@slow
				def 	__snake_case							(							self	:				Union[str, Any]      ):
										snake_case       :     Any =    logging.StreamHandler(sys.stdout      )
										logger.addHandler(SCREAMING_SNAKE_CASE_      )

										snake_case       :     Union[str, Any] =    self.get_auto_remove_tmp_dir()
										snake_case       :     Dict =    F"""
            {self.examples_dir}/pytorch/semantic-segmentation/run_semantic_segmentation_no_trainer.py
            --dataset_name huggingface/semantic-segmentation-test-sample
            --output_dir {tmp_dir}
            --max_train_steps=10
            --num_warmup_steps=2
            --learning_rate=2e-4
            --per_device_train_batch_size=2
            --per_device_eval_batch_size=1
            --checkpointing_steps epoch
        """.split()

										run_command(self._launch_args + testargs      )
										snake_case       :     List[Any] =    get_results(SCREAMING_SNAKE_CASE_      )
										self.assertGreaterEqual(result['''eval_overall_accuracy'''],    0.10      )




				@mock.patch.dict(os.environ,    {'''WANDB_MODE''': '''offline'''}      )
				def 	__snake_case							(							self	:				Union[str, Any]      ):
										snake_case       :     str =    self.get_auto_remove_tmp_dir()
										snake_case       :     Optional[Any] =    F"""
            {self.examples_dir}/pytorch/image-classification/run_image_classification_no_trainer.py
            --model_name_or_path google/vit-base-patch16-224-in21k
            --dataset_name hf-internal-testing/cats_vs_dogs_sample
            --learning_rate 1e-4
            --per_device_train_batch_size 2
            --per_device_eval_batch_size 1
            --max_train_steps 2
            --train_val_split 0.1
            --seed 42
            --output_dir {tmp_dir}
            --with_tracking
            --checkpointing_steps 1
        """.split()

										if is_cuda_and_apex_available():
																testargs.append('''--fp16'''      )

										run_command(self._launch_args + testargs      )
										snake_case       :     Tuple =    get_results(SCREAMING_SNAKE_CASE_      )
										# The base model scores a 25%
										self.assertGreaterEqual(result['''eval_accuracy'''],    0.6      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''step_1'''      )      )      )
										self.assertTrue(os.path.exists(os.path.join(SCREAMING_SNAKE_CASE_,    '''image_classification_no_trainer'''      )      )      )

