├── .clang-format ├── .github └── workflows │ └── py.yml ├── .gitignore ├── Audio ├── AM-MobileNet1D │ ├── README.md │ ├── cfg │ │ ├── AM_MobileNet1D_TIMIT.cfg │ │ └── MobileNet1D_TIMIT.cfg │ ├── infer.py │ ├── infer.sh │ ├── model │ │ └── mobilenet1d.py │ ├── requirements.txt │ ├── train.py │ ├── train.sh │ └── utils │ │ ├── TIMIT_preparation.py │ │ └── data_utils.py ├── Adaptive_Voice_Conversion │ ├── README.md │ ├── config.yaml │ ├── data_utils.py │ ├── infer.sh │ ├── inference.py │ ├── main.py │ ├── model.py │ ├── preprocess │ │ ├── make_datasets_vctk.py │ │ ├── preprocess_vctk.sh │ │ ├── reduce_dataset.py │ │ ├── sample_segments.py │ │ ├── sample_single_segments.py │ │ ├── tacotron │ │ │ ├── hyperparams.py │ │ │ └── utils.py │ │ └── vctk.config │ ├── solver.py │ ├── train.sh │ └── utils.py ├── Conv-TasNet │ ├── .gitattributes │ ├── .gitignore │ ├── README.md │ ├── conv_tasnet_loss.png │ ├── nnet │ │ ├── compute_si_snr.py │ │ ├── conv_tas_net.py │ │ ├── create_scp.py │ │ ├── libs │ │ │ ├── __init__.py │ │ │ ├── audio.py │ │ │ ├── dataset.py │ │ │ ├── metric.py │ │ │ ├── trainer.py │ │ │ └── utils.py │ │ ├── separate.py │ │ └── train.py │ ├── requirements.txt │ └── run.sh ├── CycleGAN-VC2 │ ├── README.md │ ├── infer.py │ ├── infer.sh │ ├── model │ │ ├── model.py │ │ └── trainr.py │ ├── requirements.txt │ ├── train.py │ ├── train.sh │ └── utils │ │ ├── data_preprocess.py │ │ ├── data_utils.py │ │ └── dataset.py ├── MaskCycleGAN-VC │ ├── README.md │ ├── args │ │ └── cycleGAN_arg_parser.py │ ├── infer.py │ ├── infer.sh │ ├── model │ │ ├── model.py │ │ └── trainer.py │ ├── requirements.txt │ ├── train.py │ ├── train.sh │ └── utils │ │ ├── data_preprocess.py │ │ ├── data_utils.py │ │ └── dataset.py ├── OpenTransformer │ ├── README.md │ ├── average.sh │ ├── egs │ │ └── aishell │ │ │ ├── conf │ │ │ ├── conformer_baseline.yaml │ │ │ ├── rnnlm.yaml │ │ │ ├── transformer_baseline.yaml │ │ │ └── transformer_lm.yaml │ │ │ ├── local │ │ │ ├── aishell_data_prep.sh │ │ │ ├── download_and_untar.sh │ │ │ ├── filter_scp.pl │ │ │ ├── generate_vocab.py │ │ │ ├── prepare_data.py │ │ │ └── split_and_norm.py │ │ │ ├── path.sh │ │ │ └── run.sh │ ├── eval.py │ ├── eval.sh │ ├── otrans │ │ ├── __init__.py │ │ ├── decoder │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── transformer.py │ │ │ └── utils.py │ │ ├── encoder │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── conformer.py │ │ │ ├── transformer.py │ │ │ └── utils.py │ │ ├── frontend │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── concat.py │ │ │ └── conv.py │ │ ├── model │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── ctc.py │ │ │ ├── lm.py │ │ │ └── speech2text.py │ │ ├── module │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ ├── conformer.py │ │ │ ├── ffn.py │ │ │ ├── loss.py │ │ │ └── pos.py │ │ ├── recognize │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── ctc.py │ │ │ ├── speech2text.py │ │ │ └── utils.py │ │ ├── train │ │ │ ├── __init__.py │ │ │ ├── scheduler.py │ │ │ ├── trainer.py │ │ │ └── utils.py │ │ └── utils.py │ ├── run.py │ ├── tools │ │ ├── average.py │ │ └── computer_wer.py │ └── train.sh ├── SincNet │ ├── README.md │ ├── cfg │ │ └── SincNet_TIMIT.cfg │ ├── infer.py │ ├── infer.sh │ ├── model │ │ ├── SincNet.py │ │ └── dnn_models.py │ ├── train.py │ ├── train.sh │ └── utils │ │ ├── TIMIT_preparation.py │ │ └── data_utils.py ├── Speech-Emotion-Analyzer │ ├── .editorconfig │ ├── .gitattributes │ ├── .gitignore │ ├── README.md │ ├── configs │ │ ├── cnn1d.yaml │ │ └── lstm.yaml │ ├── extract_feats │ │ ├── librosa.py │ │ └── opensmile.py │ ├── models │ │ ├── __init__.py │ │ ├── cnn1d_ser.py │ │ ├── lstm_oneflow.py │ │ └── lstm_ser.py │ ├── predict.py │ ├── predict.sh │ ├── preprocess.py │ ├── requirements.txt │ ├── train.py │ ├── train.sh │ └── utils │ │ ├── __init__.py │ │ ├── opts.py │ │ └── plot.py ├── Speech-Transformer │ ├── .gitignore │ ├── README.md │ ├── egs │ │ └── aishell │ │ │ ├── cmd.sh │ │ │ ├── conf │ │ │ └── fbank.conf │ │ │ ├── figures │ │ │ └── tr_val_loss_15000bf.png │ │ │ ├── local │ │ │ ├── aishell_data_prep.sh │ │ │ └── score.sh │ │ │ ├── loss_visualize.py │ │ │ ├── path.sh │ │ │ ├── run.sh │ │ │ ├── steps │ │ │ └── utils │ ├── src │ │ ├── __init__.py │ │ ├── bin │ │ │ ├── recognize.py │ │ │ └── train.py │ │ ├── data │ │ │ ├── __init__.py │ │ │ └── data.py │ │ ├── solver │ │ │ ├── __init__.py │ │ │ └── solver.py │ │ ├── transformer │ │ │ ├── __init__.py │ │ │ ├── attention.py │ │ │ ├── decoder.py │ │ │ ├── encoder.py │ │ │ ├── loss.py │ │ │ ├── module.py │ │ │ ├── optimizer.py │ │ │ └── transformer.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── data2json.sh │ │ │ ├── dump.sh │ │ │ ├── filt.py │ │ │ ├── json2trn.py │ │ │ ├── mergejson.py │ │ │ ├── scp2json.py │ │ │ ├── text2token.py │ │ │ └── utils.py │ └── tools │ │ ├── Makefile │ │ └── kaldi-io-for-python.tar.gz ├── StarGAN-VC │ ├── README.md │ ├── infer.sh │ ├── main.py │ ├── model │ │ ├── data_loader.py │ │ ├── model.py │ │ └── solver.py │ ├── requirements.txt │ ├── train.sh │ └── utils │ │ ├── preprocess.py │ │ └── utility.py └── Wav2Letter │ ├── README.md │ ├── Wav2Letter │ ├── data.py │ ├── decoder.py │ └── model.py │ ├── infer.py │ ├── infer.sh │ ├── requirements.txt │ ├── train.py │ └── train.sh ├── CMakeLists.txt ├── DeepReinforcementLearning ├── FlappyBird │ ├── README.md │ ├── assets │ │ └── sprites │ │ │ ├── 0.png │ │ │ ├── 1.png │ │ │ ├── 2.png │ │ │ ├── 3.png │ │ │ ├── 4.png │ │ │ ├── 5.png │ │ │ ├── 6.png │ │ │ ├── 7.png │ │ │ ├── 8.png │ │ │ ├── 9.png │ │ │ ├── background-black.png │ │ │ ├── base.png │ │ │ ├── pipe-green.png │ │ │ ├── redbird-downflap.png │ │ │ ├── redbird-midflap.png │ │ │ └── redbird-upflap.png │ ├── game │ │ ├── flappy_bird_utils.py │ │ └── wrapped_flappy_bird.py │ ├── model │ │ ├── deep_q_network.py │ │ └── utils.py │ ├── requirements.txt │ ├── test.py │ ├── test_flappy_bird.sh │ ├── train.py │ └── train_flappy_bird.sh └── MadMario │ ├── README.md │ ├── agent.py │ ├── environment.yml │ ├── main.py │ ├── metrics.py │ ├── neural.py │ ├── replay.py │ └── wrappers.py ├── Demo ├── quick_start_demo_lenet │ └── lenet.py └── speaker_identification_demo │ ├── README.md │ ├── data_preprocess.py │ ├── data_preprocess.sh │ ├── infer.py │ ├── infer.sh │ ├── model │ ├── dataloader.py │ └── model.py │ ├── train.py │ └── train.sh ├── GPT ├── examples │ ├── 2d_parallel_train.sh │ ├── pipeline_train.sh │ ├── train_117M.sh │ ├── train_2x2x2_12x1536_512x1024.sh │ └── train_2x4x4_24x2304_512x2048.sh ├── oneflow_gpt │ ├── __init__.py │ ├── config.py │ ├── data.py │ ├── distribute.py │ ├── logger.py │ ├── model.py │ ├── optimizer.py │ └── train.py └── setup.py ├── KnowledgeDistillation ├── KnowledgeDistillation │ ├── README.md │ ├── infer.py │ ├── infer.sh │ ├── model.py │ ├── requirements.txt │ ├── train.py │ └── train.sh └── MetaKD │ ├── 1_get_data.sh │ ├── 2_process_data.sh │ ├── 3_train_teacher.sh │ ├── 4_student_first.sh │ ├── 5_student_second.sh │ ├── 6_eval.sh │ ├── README.md │ ├── bert.py │ ├── bert_utils.py │ ├── easynlp │ ├── appzoo │ │ ├── __init__.py │ │ ├── api.py │ │ ├── application.py │ │ ├── dataset.py │ │ └── sequence_classification │ │ │ ├── __init__.py │ │ │ └── evaluator.py │ ├── core │ │ ├── __init__.py │ │ ├── distiller.py │ │ ├── evaluator.py │ │ ├── optimizers.py │ │ └── trainer.py │ ├── distillation │ │ ├── __init__.py │ │ ├── distill_application.py │ │ ├── distill_metakd_application.py │ │ └── distill_metakd_dataset.py │ ├── modelzoo │ │ ├── __init__.py │ │ ├── configuration_utils.py │ │ ├── file_utils.py │ │ ├── models │ │ │ ├── auto │ │ │ │ ├── __init__.py │ │ │ │ ├── configuration_auto.py │ │ │ │ └── tokenization_auto.py │ │ │ └── bert │ │ │ │ ├── __init__.py │ │ │ │ ├── configuration_bert.py │ │ │ │ ├── tokenization_bert.py │ │ │ │ └── tokenization_bert_fast.py │ │ ├── tokenization_utils.py │ │ ├── tokenization_utils_base.py │ │ ├── tokenization_utils_fast.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ └── logging.py │ ├── requirements.txt │ └── utils │ │ ├── __init__.py │ │ ├── arguments.py │ │ ├── exporter.py │ │ ├── global_vars.py │ │ ├── initializer.py │ │ ├── io_utils.py │ │ ├── logger.py │ │ ├── losses.py │ │ └── statistics.py │ ├── extract_embeddings.py │ ├── generate_dev_file.py │ ├── generate_meta_weights.py │ ├── generate_senti_data.py │ ├── main_evaluate.py │ ├── meta_student_distill.py │ ├── meta_teacher_train.py │ └── requirements.txt ├── LICENSE ├── NLP ├── BERT │ ├── README.md │ ├── config.py │ ├── examples │ │ ├── infer.sh │ │ ├── run_eager_pretrain.sh │ │ ├── run_graph_pretrain.sh │ │ └── run_squad.sh │ ├── modeling.py │ ├── run_eager_pretraining.py │ ├── run_infer.py │ ├── run_pretraining.py │ ├── run_squad.py │ ├── squad.py │ ├── squad_util.py │ ├── tokenization.py │ └── utils │ │ ├── checkpoint.py │ │ ├── comm.py │ │ ├── compare_lazy_outputs.py │ │ ├── draw_img.py │ │ ├── draw_loss_curve.py │ │ ├── lamb_optimizer.py │ │ ├── lr_scheduler.py │ │ ├── metric.py │ │ ├── ofrecord_data_utils.py │ │ ├── optimizer.py │ │ └── reporter.py ├── CPT │ ├── README.md │ ├── classifier_flow.py │ ├── dataset_flow.py │ ├── infer.sh │ ├── infer_flow.py │ ├── models │ │ ├── CPT.py │ │ ├── bart_utils.py │ │ ├── bert.py │ │ ├── bert_utils.py │ │ └── utils.py │ ├── requirements.txt │ ├── train.sh │ └── train_flow.py ├── GPT2 │ ├── README.md │ ├── compare_speed.py │ ├── config.py │ ├── convert_pt_ckpt_to_of.py │ ├── finetune.sh │ ├── generate_sample.py │ ├── gpt_dataset.py │ ├── gpt_dataset_pt.py │ ├── infer.sh │ ├── model.py │ ├── model_config.py │ ├── optimization.py │ ├── optimization_pt.py │ ├── pt_model.py │ ├── requirements.txt │ ├── run_generation.py │ ├── tokenizer │ │ ├── __init__.py │ │ ├── gpt2_tokenization.py │ │ └── tokenizer.py │ ├── train.py │ ├── train.sh │ ├── train_pt.py │ ├── trainer.py │ └── trainer_pt.py ├── LSTMText │ ├── README.md │ ├── infer.py │ ├── infer.sh │ ├── model.py │ ├── requirements.txt │ ├── train.sh │ ├── train_bilstm.py │ └── utils.py ├── MoE │ ├── README.md │ ├── cifar10_example.py │ ├── mlp.py │ ├── moe.py │ ├── requirements.txt │ └── train.sh ├── TextCNN │ ├── README.md │ ├── infer.py │ ├── infer.sh │ ├── model.py │ ├── requirements.txt │ ├── train.sh │ ├── train_of_textcnn.py │ ├── training.py │ └── utils.py ├── Transformer │ ├── README.md │ ├── imdb │ │ ├── infer.sh │ │ ├── infer_transformer_imdb.py │ │ ├── model.py │ │ ├── train.sh │ │ ├── train_transformer_imdb.py │ │ └── utils.py │ ├── odd_numbers │ │ ├── infer.sh │ │ ├── infer_transformer_odd_numbers.py │ │ ├── model.py │ │ ├── train.sh │ │ └── train_transformer_odd_numbers.py │ └── transformer │ │ ├── __init__.py │ │ ├── multihead_attention.py │ │ ├── transformer.py │ │ └── utils.py ├── rnn │ ├── README.md │ ├── compare_oneflow_and_pytorch_lstm_speed.py │ ├── compare_oneflow_and_pytorch_rnn_speed.py │ ├── models │ │ ├── __init__.py │ │ ├── lstm_oneflow.py │ │ ├── lstm_pytorch.py │ │ ├── rnn_model.py │ │ └── rnn_model_pytorch.py │ ├── predict.py │ ├── requirements.txt │ ├── train.sh │ ├── train_lstm_oneflow.py │ ├── train_rnn_oneflow.py │ └── utils │ │ ├── __init__.py │ │ ├── dataset.py │ │ └── tensor_utils.py ├── roberta │ ├── MNLIDataset.py │ ├── README.md │ ├── SST2Dataset.py │ ├── classifier_MNLI.py │ ├── classifier_SST2.py │ ├── config.py │ ├── infer_MNLI.py │ ├── infer_MNLI.sh │ ├── infer_SST2.py │ ├── infer_SST2.sh │ ├── models │ │ ├── roberta.py │ │ └── roberta_utils.py │ ├── requirements.txt │ ├── train_MNLI.py │ ├── train_MNLI.sh │ ├── train_SST2.py │ └── train_SST2.sh └── seq2seq │ ├── README.md │ ├── eval_oneflow.py │ ├── eval_oneflow.sh │ ├── models │ ├── GRU_oneflow.py │ ├── __init__.py │ └── seq_seq_oneflow.py │ ├── requirements.txt │ ├── train_oneflow.py │ ├── train_oneflow.sh │ └── utils │ ├── __init__.py │ ├── dataset.py │ └── utils_oneflow.py ├── Quantization ├── README.md ├── __init__.py ├── infer.sh ├── models │ └── q_alexnet.py ├── quantization_aware_training.py ├── quantization_infer.py ├── quantization_ops │ ├── __init__.py │ ├── conv.py │ ├── conv_bn.py │ ├── linear.py │ └── q_module.py ├── train.sh └── utils │ ├── __init__.py │ ├── imagenet1000_clsidx_to_labels.py │ ├── numpy_data_utils.py │ ├── ofrecord_data_utils.py │ └── plot.py ├── README.md ├── README_zh-CN.md ├── RecommenderSystems ├── dcn │ ├── README.md │ ├── dcn_eager_train_eval.py │ ├── dcn_train_eval.py │ ├── eager_train.sh │ ├── requirements.txt │ ├── tools │ │ ├── dcn_parquet.scala │ │ ├── launch_spark.sh │ │ └── split_criteo.py │ └── train.sh ├── deepfm │ ├── README.md │ ├── deepfm_eager_train_eval.py │ ├── deepfm_train_eval.py │ ├── eager_train.sh │ ├── tools │ │ ├── deepfm_parquet.scala │ │ ├── launch_spark.sh │ │ └── split_criteo_kaggle.py │ └── train_deepfm.sh ├── dlrm │ ├── README.md │ ├── __init__.py │ ├── dlrm_train_eval.py │ ├── requirements.txt │ └── tools │ │ ├── criteo1t_parquet.py │ │ ├── criteo1t_parquet.scala │ │ ├── criteo1t_parquet_day_by_day.scala │ │ ├── launch_spark.sh │ │ └── split_day_23.sh ├── mmoe │ ├── README.md │ ├── mmoe_train_eval.py │ ├── tools │ │ └── mmoe_parquet.py │ └── train_mmoe.sh ├── pnn │ ├── README.md │ ├── pnn_train_eval.py │ ├── tools │ │ ├── criteo_parquet.py │ │ ├── launch_spark.sh │ │ └── split_criteo_kaggle.py │ └── train_pnn.sh ├── wide_and_deep │ ├── README.md │ ├── __init__.py │ ├── config.py │ ├── graph.py │ ├── models │ │ ├── __init__.py │ │ ├── data.py │ │ └── wide_and_deep.py │ ├── train.py │ ├── train_ddp.sh │ ├── train_global_eager.sh │ ├── train_global_graph.sh │ ├── train_nn_graph.sh │ └── utils │ │ └── logger.py └── xdeepfm │ ├── README.md │ ├── tools │ ├── launch_spark.sh │ ├── split_criteo_kaggle.py │ └── xdeepfm_parquet.scala │ ├── train_xdeepfm.sh │ └── xdeepfm_train_eval.py ├── Vision ├── BJTU_summer_tournament │ ├── DataSet.py │ ├── DenseNet.py │ ├── Predict.py │ ├── Prepare.py │ ├── README.md │ ├── Train.py │ └── requirements.txt ├── LSTM │ └── EnResLSTM │ │ ├── README.md │ │ ├── infer.py │ │ ├── run.sh │ │ └── train.py ├── classification │ ├── image │ │ ├── DLA │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ └── dla.py │ │ │ ├── requirements.txt │ │ │ ├── train_oneflow.py │ │ │ ├── train_oneflow.sh │ │ │ └── utils │ │ │ │ ├── clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ └── ofrecord_data_utils.py │ │ ├── ViT │ │ │ ├── README.md │ │ │ ├── eager │ │ │ │ ├── infer.py │ │ │ │ ├── infer.sh │ │ │ │ ├── train.py │ │ │ │ └── train.sh │ │ │ ├── graph │ │ │ │ ├── infer.py │ │ │ │ ├── infer.sh │ │ │ │ ├── train.py │ │ │ │ └── train.sh │ │ │ ├── model │ │ │ │ ├── __init__.py │ │ │ │ ├── build_model.py │ │ │ │ └── vit.py │ │ │ ├── requirements.txt │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ └── ofrecord_data_utils.py │ │ ├── alexnet │ │ │ ├── README.md │ │ │ ├── check │ │ │ │ ├── check.py │ │ │ │ ├── check.sh │ │ │ │ ├── draw.py │ │ │ │ └── draw.sh │ │ │ ├── eager │ │ │ │ ├── infer.py │ │ │ │ ├── infer.sh │ │ │ │ ├── train.py │ │ │ │ └── train.sh │ │ │ ├── graph │ │ │ │ ├── infer.py │ │ │ │ ├── infer.sh │ │ │ │ ├── train.py │ │ │ │ └── train.sh │ │ │ ├── model │ │ │ │ ├── __init__.py │ │ │ │ └── alexnet.py │ │ │ ├── requirements.txt │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ └── plot.py │ │ ├── densenet │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ └── densenet.py │ │ │ ├── requirements.txt │ │ │ ├── train.py │ │ │ ├── train.sh │ │ │ └── utils │ │ │ │ ├── Figure_1.png │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ └── plot.py │ │ ├── ghostnet │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ └── ghostnet.py │ │ │ ├── train_oneflow.py │ │ │ ├── train_oneflow.sh │ │ │ └── utils │ │ │ │ ├── ghostnet_compare.png │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ └── ofrecord_data_utils.py │ │ ├── inception_v3 │ │ │ ├── README.md │ │ │ ├── eager │ │ │ │ ├── infer.py │ │ │ │ ├── infer.sh │ │ │ │ ├── train.py │ │ │ │ └── train.sh │ │ │ ├── graph │ │ │ │ ├── build_graph.py │ │ │ │ ├── config.py │ │ │ │ ├── data.py │ │ │ │ ├── infer.py │ │ │ │ ├── infer.sh │ │ │ │ ├── train.py │ │ │ │ └── train_ddp.sh │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ └── inceptionv3.py │ │ │ ├── requirements.txt │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ └── plot.py │ │ ├── mnasnet │ │ │ ├── README.md │ │ │ ├── eager │ │ │ │ ├── infer.py │ │ │ │ ├── infer.sh │ │ │ │ ├── train.py │ │ │ │ └── train.sh │ │ │ ├── graph │ │ │ │ ├── infer.py │ │ │ │ ├── infer.sh │ │ │ │ ├── train.py │ │ │ │ └── train.sh │ │ │ ├── model │ │ │ │ ├── __init__.py │ │ │ │ └── mnasnet.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ └── plot.py │ │ ├── mobilenetv2 │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ └── mobilenetv2.py │ │ │ ├── requirements.txt │ │ │ ├── train.py │ │ │ ├── train.sh │ │ │ └── utils │ │ │ │ ├── Figure_1.png │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ └── plot.py │ │ ├── mobilenetv3 │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ └── mobilenetv3.py │ │ │ ├── requirements.txt │ │ │ ├── train.py │ │ │ ├── train.sh │ │ │ └── utils │ │ │ │ ├── Figure_1.png │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ └── plot.py │ │ ├── poseNet │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ └── posenet.py │ │ │ ├── requirements.txt │ │ │ ├── train_oneflow.py │ │ │ ├── train_oneflow.sh │ │ │ └── utils │ │ │ │ ├── clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ └── ofrecord_data_utils.py │ │ ├── repvgg │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ └── repvgg.py │ │ │ ├── requirements.txt │ │ │ ├── train.py │ │ │ ├── train.sh │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ └── plot.py │ │ ├── resnet50 │ │ │ ├── README.md │ │ │ ├── check │ │ │ │ ├── check.py │ │ │ │ ├── check.sh │ │ │ │ ├── draw.py │ │ │ │ └── draw.sh │ │ │ ├── config.py │ │ │ ├── examples │ │ │ │ ├── infer.sh │ │ │ │ ├── infer_graph.sh │ │ │ │ ├── train_ddp_fp32.sh │ │ │ │ ├── train_ddp_stat.sh │ │ │ │ ├── train_eager.sh │ │ │ │ ├── train_eager_consistent.sh │ │ │ │ ├── train_graph.sh │ │ │ │ ├── train_graph_distributed_fp16.sh │ │ │ │ └── train_graph_distributed_fp32.sh │ │ │ ├── graph.py │ │ │ ├── infer.py │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ ├── accuracy.py │ │ │ │ ├── data.py │ │ │ │ ├── optimizer.py │ │ │ │ ├── pytorch_resnet50.py │ │ │ │ └── resnet50.py │ │ │ ├── requirements.txt │ │ │ ├── train.py │ │ │ └── utils │ │ │ │ ├── Figure_1.png │ │ │ │ ├── __init__.py │ │ │ │ ├── debug.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── logger.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ ├── plot.py │ │ │ │ ├── printer.py │ │ │ │ └── stat.py │ │ ├── resnext50_32x4d │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ ├── __init__.py │ │ │ │ └── resnext50_32x4d.py │ │ │ ├── requirements.txt │ │ │ ├── train.py │ │ │ ├── train.sh │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ └── plot.py │ │ ├── scnet │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ ├── resnet50.py │ │ │ │ └── scloss.py │ │ │ ├── requirements.txt │ │ │ ├── train.py │ │ │ ├── train.sh │ │ │ └── utils │ │ │ │ ├── clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ └── ofrecord_data_utils.py │ │ ├── shufflenetv2 │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ └── shufflenetv2.py │ │ │ ├── requirements.txt │ │ │ ├── train.py │ │ │ ├── train.sh │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ └── plot.py │ │ ├── swin_transformer │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ │ └── swin_transformer.py │ │ │ ├── requirements.txt │ │ │ ├── train.py │ │ │ ├── train.sh │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ │ ├── numpy_data_utils.py │ │ │ │ ├── ofrecord_data_utils.py │ │ │ │ └── plot.py │ │ └── vgg │ │ │ ├── README.md │ │ │ ├── infer.py │ │ │ ├── infer.sh │ │ │ ├── models │ │ │ ├── __init__.py │ │ │ └── vgg.py │ │ │ ├── train_oneflow.py │ │ │ ├── train_oneflow.sh │ │ │ └── utils │ │ │ ├── Figure_1.png │ │ │ ├── __init__.py │ │ │ ├── imagenet1000_clsidx_to_labels.py │ │ │ ├── numpy_data_utils.py │ │ │ ├── ofrecord_data_utils.py │ │ │ └── plot.py │ └── video │ │ └── TSN │ │ ├── README.md │ │ ├── infer.sh │ │ ├── requirements.txt │ │ ├── test_recognizer.py │ │ ├── train.sh │ │ ├── train_recognizer.py │ │ └── tsn │ │ ├── datasets │ │ ├── dataset.py │ │ └── transform.py │ │ ├── models │ │ ├── TSN.py │ │ ├── cls_head.py │ │ ├── resnet50.py │ │ ├── simple_consensus.py │ │ └── simple_spatial_module.py │ │ └── utils │ │ └── checkpoint.py ├── detection │ └── CSRNet │ │ ├── README.md │ │ ├── dataset.py │ │ ├── image.py │ │ ├── infer.py │ │ ├── infer.sh │ │ ├── model.py │ │ ├── part_A_test.json │ │ ├── part_A_train.json │ │ ├── part_A_train_with_val.json │ │ ├── part_A_val.json │ │ ├── part_B_test.json │ │ ├── part_B_train.json │ │ ├── part_B_train_with_val.json │ │ ├── part_B_val.json │ │ ├── requirement.txt │ │ ├── train.py │ │ ├── train.sh │ │ ├── transforms │ │ ├── spatial_transforms.py │ │ └── temporal_transforms.py │ │ ├── utils.py │ │ ├── val.py │ │ ├── val.sh │ │ └── vgg.py ├── gan │ ├── DCGAN │ │ ├── README.md │ │ ├── check │ │ │ ├── check.py │ │ │ └── check.sh │ │ ├── eager │ │ │ ├── train.py │ │ │ └── train.sh │ │ ├── graph │ │ │ ├── train.py │ │ │ └── train.sh │ │ ├── models.py │ │ ├── plot.py │ │ ├── test │ │ │ ├── test_of_dcgan.py │ │ │ ├── test_of_dcgan.sh │ │ │ └── test_of_dcgan_graph.sh │ │ ├── test_images.png │ │ └── utils.py │ ├── SRGAN │ │ ├── README.md │ │ ├── models │ │ │ └── of_model.py │ │ ├── requirements.txt │ │ ├── results.png │ │ ├── test_of_srgan.py │ │ ├── test_of_srgan.sh │ │ ├── train_of_srgan.py │ │ ├── train_of_srgan.sh │ │ ├── utils │ │ │ ├── of_data_utils.py │ │ │ └── of_loss.py │ │ └── vgg.py │ ├── cycleGAN │ │ ├── README.md │ │ ├── cycleGAN.py │ │ ├── download.sh │ │ ├── image.py │ │ ├── infer.py │ │ ├── infer.sh │ │ ├── networks.py │ │ ├── requirements.txt │ │ ├── train.py │ │ └── train.sh │ └── pix2pix │ │ ├── README.md │ │ ├── infer.py │ │ ├── infer.sh │ │ ├── models │ │ └── networks.py │ │ ├── requirements.txt │ │ ├── train.py │ │ ├── train.sh │ │ └── utils │ │ ├── data_utils.py │ │ ├── plot.py │ │ └── utils.py ├── oneflow_face │ ├── README.md │ ├── README_CH.md │ ├── backbones │ │ ├── __init__.py │ │ └── ir_resnet.py │ ├── configs │ │ ├── __init__.py │ │ ├── base.py │ │ ├── glint360k_mbf.py │ │ ├── glint360k_r100.py │ │ ├── glint360k_r18.py │ │ ├── glint360k_r34.py │ │ ├── glint360k_r50.py │ │ ├── ms1mv3_mbf.py │ │ ├── ms1mv3_r18.py │ │ ├── ms1mv3_r34.py │ │ ├── ms1mv3_r50.py │ │ └── speed.py │ ├── convert.sh │ ├── eval │ │ ├── __init__.py │ │ ├── onnx_helper.py │ │ ├── onnx_ijbc.py │ │ └── verification.py │ ├── function.py │ ├── graph.py │ ├── oneflow2onnx.py │ ├── requirements.txt │ ├── tools │ │ ├── mx_recordio_2_ofrecord.py │ │ └── mx_recordio_2_ofrecord_shuffled_npart.py │ ├── train.py │ ├── train_ddp.sh │ ├── train_graph_distributed.sh │ ├── utils │ │ ├── __init__.py │ │ ├── losses.py │ │ ├── ofrecord_data_utils.py │ │ ├── utils_callbacks.py │ │ ├── utils_config.py │ │ └── utils_logging.py │ ├── val.py │ └── val.sh ├── reid │ └── BoT │ │ ├── README.md │ │ ├── data_loader.py │ │ ├── infer.sh │ │ ├── loss.py │ │ ├── lr_scheduler.py │ │ ├── model.py │ │ ├── reid.py │ │ ├── requirements.txt │ │ ├── train.sh │ │ └── utils │ │ ├── __init__.py │ │ ├── distance.py │ │ └── loggers.py ├── segmentation │ ├── FODDet │ │ ├── README.md │ │ ├── model │ │ │ └── UNet.py │ │ ├── requirements.txt │ │ ├── test │ │ │ ├── dataloader.py │ │ │ ├── test.py │ │ │ ├── test.sh │ │ │ └── visualize.py │ │ └── train │ │ │ ├── train.py │ │ │ └── train.sh │ ├── FaceSeg │ │ ├── README.md │ │ ├── infer.py │ │ ├── infer.sh │ │ ├── models │ │ │ ├── LinkNet34.py │ │ │ ├── __init__.py │ │ │ └── resnet50.py │ │ ├── requirements.txt │ │ ├── train.py │ │ ├── train.sh │ │ ├── utils │ │ │ ├── __init__.py │ │ │ └── numpy_data_utils.py │ │ └── validation_oneflow.py │ └── U-Net │ │ ├── README.md │ │ ├── TrainUnetDataSet.py │ │ ├── dataloader.py │ │ ├── log.py │ │ ├── plot.py │ │ ├── predict_unet_test.py │ │ ├── requirements.txt │ │ ├── test.sh │ │ ├── train.sh │ │ └── unet.py └── style_transform │ └── fast_neural_style │ ├── README.md │ ├── infer.sh │ ├── neural_style │ ├── neural_style.py │ ├── transformer_net.py │ ├── utils.py │ └── vgg.py │ ├── requirements.txt │ └── train.sh ├── ci └── check │ └── run_py_format.py ├── dev-requirements.txt ├── ops ├── __init__.py ├── csrc │ ├── CMakeLists.txt │ ├── nms.cpp │ ├── nms.cu │ ├── roi_align.cpp │ ├── roi_align.cu │ └── roi_align.h ├── nms.py ├── roi_align.py └── test │ ├── nms_test.py │ └── roi_align_test.py └── scripts ├── compare_speed_with_pytorch.py └── swin_dataloader_compare_speed_with_pytorch.py /.clang-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/.clang-format -------------------------------------------------------------------------------- /.github/workflows/py.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/.github/workflows/py.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/.gitignore -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/AM-MobileNet1D/README.md -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/cfg/AM_MobileNet1D_TIMIT.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/AM-MobileNet1D/cfg/AM_MobileNet1D_TIMIT.cfg -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/cfg/MobileNet1D_TIMIT.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/AM-MobileNet1D/cfg/MobileNet1D_TIMIT.cfg -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/AM-MobileNet1D/infer.py -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/AM-MobileNet1D/infer.sh -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/model/mobilenet1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/AM-MobileNet1D/model/mobilenet1d.py -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/requirements.txt: -------------------------------------------------------------------------------- 1 | SoundFile 2 | tqdm 3 | oneflow 4 | numpy 5 | -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/AM-MobileNet1D/train.py -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/AM-MobileNet1D/train.sh -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/utils/TIMIT_preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/AM-MobileNet1D/utils/TIMIT_preparation.py -------------------------------------------------------------------------------- /Audio/AM-MobileNet1D/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/AM-MobileNet1D/utils/data_utils.py -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/README.md -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/config.yaml -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/data_utils.py -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/infer.sh -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/inference.py -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/main.py -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/model.py -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/preprocess/vctk.config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/preprocess/vctk.config -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/solver.py -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/train.sh -------------------------------------------------------------------------------- /Audio/Adaptive_Voice_Conversion/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Adaptive_Voice_Conversion/utils.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/.gitattributes -------------------------------------------------------------------------------- /Audio/Conv-TasNet/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/.gitignore -------------------------------------------------------------------------------- /Audio/Conv-TasNet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/README.md -------------------------------------------------------------------------------- /Audio/Conv-TasNet/conv_tasnet_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/conv_tasnet_loss.png -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/compute_si_snr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/nnet/compute_si_snr.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/conv_tas_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/nnet/conv_tas_net.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/create_scp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/nnet/create_scp.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/libs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/libs/audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/nnet/libs/audio.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/libs/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/nnet/libs/dataset.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/libs/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/nnet/libs/metric.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/libs/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/nnet/libs/trainer.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/libs/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/nnet/libs/utils.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/separate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/nnet/separate.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/nnet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/nnet/train.py -------------------------------------------------------------------------------- /Audio/Conv-TasNet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/requirements.txt -------------------------------------------------------------------------------- /Audio/Conv-TasNet/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Conv-TasNet/run.sh -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/CycleGAN-VC2/README.md -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/CycleGAN-VC2/infer.py -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/CycleGAN-VC2/infer.sh -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/CycleGAN-VC2/model/model.py -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/model/trainr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/CycleGAN-VC2/model/trainr.py -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/requirements.txt: -------------------------------------------------------------------------------- 1 | oneflow 2 | librosa 3 | tqdm 4 | pyworld 5 | -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/CycleGAN-VC2/train.py -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/CycleGAN-VC2/train.sh -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/utils/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/CycleGAN-VC2/utils/data_preprocess.py -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/CycleGAN-VC2/utils/data_utils.py -------------------------------------------------------------------------------- /Audio/CycleGAN-VC2/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/CycleGAN-VC2/utils/dataset.py -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/README.md -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/args/cycleGAN_arg_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/args/cycleGAN_arg_parser.py -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/infer.py -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/infer.sh -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/model/model.py -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/model/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/model/trainer.py -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/requirements.txt: -------------------------------------------------------------------------------- 1 | oneflow 2 | librosa 3 | tqdm 4 | pyworld -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/train.py -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/train.sh -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/utils/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/utils/data_preprocess.py -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/utils/data_utils.py -------------------------------------------------------------------------------- /Audio/MaskCycleGAN-VC/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/MaskCycleGAN-VC/utils/dataset.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/README.md -------------------------------------------------------------------------------- /Audio/OpenTransformer/average.sh: -------------------------------------------------------------------------------- 1 | python3 tools/average.py egs/aishell/exp/transformer_baseline 70 79 -------------------------------------------------------------------------------- /Audio/OpenTransformer/egs/aishell/conf/rnnlm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/egs/aishell/conf/rnnlm.yaml -------------------------------------------------------------------------------- /Audio/OpenTransformer/egs/aishell/local/filter_scp.pl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/egs/aishell/local/filter_scp.pl -------------------------------------------------------------------------------- /Audio/OpenTransformer/egs/aishell/local/generate_vocab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/egs/aishell/local/generate_vocab.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/egs/aishell/local/prepare_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/egs/aishell/local/prepare_data.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/egs/aishell/local/split_and_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/egs/aishell/local/split_and_norm.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/egs/aishell/path.sh: -------------------------------------------------------------------------------- 1 | KALDI_ROOT='~/kaldi' -------------------------------------------------------------------------------- /Audio/OpenTransformer/egs/aishell/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/egs/aishell/run.sh -------------------------------------------------------------------------------- /Audio/OpenTransformer/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/eval.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/eval.sh -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/decoder/__init__.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/decoder/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/decoder/base.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/decoder/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/decoder/transformer.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/decoder/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/decoder/utils.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/encoder/__init__.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/encoder/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/encoder/base.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/encoder/conformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/encoder/conformer.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/encoder/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/encoder/transformer.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/encoder/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/encoder/utils.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/frontend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/frontend/__init__.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/frontend/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/frontend/base.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/frontend/concat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/frontend/concat.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/frontend/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/frontend/conv.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/model/__init__.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/model/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/model/base.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/model/ctc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/model/ctc.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/model/lm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/model/lm.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/model/speech2text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/model/speech2text.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/module/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/module/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/module/attention.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/module/conformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/module/conformer.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/module/ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/module/ffn.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/module/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/module/loss.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/module/pos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/module/pos.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/recognize/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/recognize/__init__.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/recognize/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/recognize/base.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/recognize/ctc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/recognize/ctc.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/recognize/speech2text.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/recognize/speech2text.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/recognize/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/recognize/utils.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/train/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/train/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/train/scheduler.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/train/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/train/trainer.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/train/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/train/utils.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/otrans/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/otrans/utils.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/run.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/tools/average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/tools/average.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/tools/computer_wer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/tools/computer_wer.py -------------------------------------------------------------------------------- /Audio/OpenTransformer/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/OpenTransformer/train.sh -------------------------------------------------------------------------------- /Audio/SincNet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/SincNet/README.md -------------------------------------------------------------------------------- /Audio/SincNet/cfg/SincNet_TIMIT.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/SincNet/cfg/SincNet_TIMIT.cfg -------------------------------------------------------------------------------- /Audio/SincNet/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/SincNet/infer.py -------------------------------------------------------------------------------- /Audio/SincNet/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/SincNet/infer.sh -------------------------------------------------------------------------------- /Audio/SincNet/model/SincNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/SincNet/model/SincNet.py -------------------------------------------------------------------------------- /Audio/SincNet/model/dnn_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/SincNet/model/dnn_models.py -------------------------------------------------------------------------------- /Audio/SincNet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/SincNet/train.py -------------------------------------------------------------------------------- /Audio/SincNet/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/SincNet/train.sh -------------------------------------------------------------------------------- /Audio/SincNet/utils/TIMIT_preparation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/SincNet/utils/TIMIT_preparation.py -------------------------------------------------------------------------------- /Audio/SincNet/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/SincNet/utils/data_utils.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/.editorconfig: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/.editorconfig -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/.gitattributes -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/.gitignore -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/README.md -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/configs/cnn1d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/configs/cnn1d.yaml -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/configs/lstm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/configs/lstm.yaml -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/extract_feats/librosa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/extract_feats/librosa.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/extract_feats/opensmile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/extract_feats/opensmile.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/models/__init__.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/models/cnn1d_ser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/models/cnn1d_ser.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/models/lstm_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/models/lstm_oneflow.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/models/lstm_ser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/models/lstm_ser.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/predict.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/predict.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/predict.sh -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/preprocess.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/requirements.txt -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/train.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/train.sh -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/utils/__init__.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/utils/opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/utils/opts.py -------------------------------------------------------------------------------- /Audio/Speech-Emotion-Analyzer/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Emotion-Analyzer/utils/plot.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/.gitignore -------------------------------------------------------------------------------- /Audio/Speech-Transformer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/README.md -------------------------------------------------------------------------------- /Audio/Speech-Transformer/egs/aishell/cmd.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/egs/aishell/cmd.sh -------------------------------------------------------------------------------- /Audio/Speech-Transformer/egs/aishell/conf/fbank.conf: -------------------------------------------------------------------------------- 1 | --sample-frequency=16000 2 | --num-mel-bins=80 -------------------------------------------------------------------------------- /Audio/Speech-Transformer/egs/aishell/local/score.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/egs/aishell/local/score.sh -------------------------------------------------------------------------------- /Audio/Speech-Transformer/egs/aishell/loss_visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/egs/aishell/loss_visualize.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/egs/aishell/path.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/egs/aishell/path.sh -------------------------------------------------------------------------------- /Audio/Speech-Transformer/egs/aishell/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/egs/aishell/run.sh -------------------------------------------------------------------------------- /Audio/Speech-Transformer/egs/aishell/steps: -------------------------------------------------------------------------------- 1 | ../../tools/kaldi/egs/wsj/s5/steps -------------------------------------------------------------------------------- /Audio/Speech-Transformer/egs/aishell/utils: -------------------------------------------------------------------------------- 1 | ../../tools/kaldi/egs/wsj/s5/utils -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/bin/recognize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/bin/recognize.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/bin/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/bin/train.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/data/__init__.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/data/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/data/data.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/solver/__init__.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/solver/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/solver/solver.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/transformer/__init__.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/transformer/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/transformer/attention.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/transformer/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/transformer/decoder.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/transformer/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/transformer/encoder.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/transformer/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/transformer/loss.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/transformer/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/transformer/module.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/transformer/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/transformer/optimizer.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/utils/__init__.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/utils/data2json.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/utils/data2json.sh -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/utils/dump.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/utils/dump.sh -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/utils/filt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/utils/filt.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/utils/json2trn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/utils/json2trn.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/utils/mergejson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/utils/mergejson.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/utils/scp2json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/utils/scp2json.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/utils/text2token.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/utils/text2token.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/src/utils/utils.py -------------------------------------------------------------------------------- /Audio/Speech-Transformer/tools/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Speech-Transformer/tools/Makefile -------------------------------------------------------------------------------- /Audio/StarGAN-VC/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/StarGAN-VC/README.md -------------------------------------------------------------------------------- /Audio/StarGAN-VC/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/StarGAN-VC/infer.sh -------------------------------------------------------------------------------- /Audio/StarGAN-VC/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/StarGAN-VC/main.py -------------------------------------------------------------------------------- /Audio/StarGAN-VC/model/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/StarGAN-VC/model/data_loader.py -------------------------------------------------------------------------------- /Audio/StarGAN-VC/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/StarGAN-VC/model/model.py -------------------------------------------------------------------------------- /Audio/StarGAN-VC/model/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/StarGAN-VC/model/solver.py -------------------------------------------------------------------------------- /Audio/StarGAN-VC/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | librosa 3 | oneflow 4 | SoundFile 5 | pyworld 6 | scikit_learn 7 | -------------------------------------------------------------------------------- /Audio/StarGAN-VC/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/StarGAN-VC/train.sh -------------------------------------------------------------------------------- /Audio/StarGAN-VC/utils/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/StarGAN-VC/utils/preprocess.py -------------------------------------------------------------------------------- /Audio/StarGAN-VC/utils/utility.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/StarGAN-VC/utils/utility.py -------------------------------------------------------------------------------- /Audio/Wav2Letter/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Wav2Letter/README.md -------------------------------------------------------------------------------- /Audio/Wav2Letter/Wav2Letter/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Wav2Letter/Wav2Letter/data.py -------------------------------------------------------------------------------- /Audio/Wav2Letter/Wav2Letter/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Wav2Letter/Wav2Letter/decoder.py -------------------------------------------------------------------------------- /Audio/Wav2Letter/Wav2Letter/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Wav2Letter/Wav2Letter/model.py -------------------------------------------------------------------------------- /Audio/Wav2Letter/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Wav2Letter/infer.py -------------------------------------------------------------------------------- /Audio/Wav2Letter/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Wav2Letter/infer.sh -------------------------------------------------------------------------------- /Audio/Wav2Letter/requirements.txt: -------------------------------------------------------------------------------- 1 | oneflow 2 | sonopy 3 | scipy 4 | tqdm 5 | Levenshtein 6 | -------------------------------------------------------------------------------- /Audio/Wav2Letter/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Wav2Letter/train.py -------------------------------------------------------------------------------- /Audio/Wav2Letter/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Audio/Wav2Letter/train.sh -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- 1 | cmake_minimum_required(VERSION 3.0) 2 | project(models) 3 | add_subdirectory(ops/csrc) 4 | -------------------------------------------------------------------------------- /DeepReinforcementLearning/FlappyBird/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/FlappyBird/README.md -------------------------------------------------------------------------------- /DeepReinforcementLearning/FlappyBird/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/FlappyBird/model/utils.py -------------------------------------------------------------------------------- /DeepReinforcementLearning/FlappyBird/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | pygame 3 | opencv_python -------------------------------------------------------------------------------- /DeepReinforcementLearning/FlappyBird/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/FlappyBird/test.py -------------------------------------------------------------------------------- /DeepReinforcementLearning/FlappyBird/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/FlappyBird/train.py -------------------------------------------------------------------------------- /DeepReinforcementLearning/MadMario/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/MadMario/README.md -------------------------------------------------------------------------------- /DeepReinforcementLearning/MadMario/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/MadMario/agent.py -------------------------------------------------------------------------------- /DeepReinforcementLearning/MadMario/environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/MadMario/environment.yml -------------------------------------------------------------------------------- /DeepReinforcementLearning/MadMario/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/MadMario/main.py -------------------------------------------------------------------------------- /DeepReinforcementLearning/MadMario/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/MadMario/metrics.py -------------------------------------------------------------------------------- /DeepReinforcementLearning/MadMario/neural.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/MadMario/neural.py -------------------------------------------------------------------------------- /DeepReinforcementLearning/MadMario/replay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/MadMario/replay.py -------------------------------------------------------------------------------- /DeepReinforcementLearning/MadMario/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/DeepReinforcementLearning/MadMario/wrappers.py -------------------------------------------------------------------------------- /Demo/quick_start_demo_lenet/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Demo/quick_start_demo_lenet/lenet.py -------------------------------------------------------------------------------- /Demo/speaker_identification_demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Demo/speaker_identification_demo/README.md -------------------------------------------------------------------------------- /Demo/speaker_identification_demo/data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Demo/speaker_identification_demo/data_preprocess.py -------------------------------------------------------------------------------- /Demo/speaker_identification_demo/data_preprocess.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Demo/speaker_identification_demo/data_preprocess.sh -------------------------------------------------------------------------------- /Demo/speaker_identification_demo/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Demo/speaker_identification_demo/infer.py -------------------------------------------------------------------------------- /Demo/speaker_identification_demo/infer.sh: -------------------------------------------------------------------------------- 1 | python infer.py 2 | -------------------------------------------------------------------------------- /Demo/speaker_identification_demo/model/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Demo/speaker_identification_demo/model/dataloader.py -------------------------------------------------------------------------------- /Demo/speaker_identification_demo/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Demo/speaker_identification_demo/model/model.py -------------------------------------------------------------------------------- /Demo/speaker_identification_demo/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Demo/speaker_identification_demo/train.py -------------------------------------------------------------------------------- /Demo/speaker_identification_demo/train.sh: -------------------------------------------------------------------------------- 1 | python train.py 2 | -------------------------------------------------------------------------------- /GPT/examples/2d_parallel_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/examples/2d_parallel_train.sh -------------------------------------------------------------------------------- /GPT/examples/pipeline_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/examples/pipeline_train.sh -------------------------------------------------------------------------------- /GPT/examples/train_117M.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/examples/train_117M.sh -------------------------------------------------------------------------------- /GPT/examples/train_2x2x2_12x1536_512x1024.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/examples/train_2x2x2_12x1536_512x1024.sh -------------------------------------------------------------------------------- /GPT/examples/train_2x4x4_24x2304_512x2048.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/examples/train_2x4x4_24x2304_512x2048.sh -------------------------------------------------------------------------------- /GPT/oneflow_gpt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/oneflow_gpt/__init__.py -------------------------------------------------------------------------------- /GPT/oneflow_gpt/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/oneflow_gpt/config.py -------------------------------------------------------------------------------- /GPT/oneflow_gpt/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/oneflow_gpt/data.py -------------------------------------------------------------------------------- /GPT/oneflow_gpt/distribute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/oneflow_gpt/distribute.py -------------------------------------------------------------------------------- /GPT/oneflow_gpt/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/oneflow_gpt/logger.py -------------------------------------------------------------------------------- /GPT/oneflow_gpt/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/oneflow_gpt/model.py -------------------------------------------------------------------------------- /GPT/oneflow_gpt/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/oneflow_gpt/optimizer.py -------------------------------------------------------------------------------- /GPT/oneflow_gpt/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/oneflow_gpt/train.py -------------------------------------------------------------------------------- /GPT/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/GPT/setup.py -------------------------------------------------------------------------------- /KnowledgeDistillation/KnowledgeDistillation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/KnowledgeDistillation/README.md -------------------------------------------------------------------------------- /KnowledgeDistillation/KnowledgeDistillation/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/KnowledgeDistillation/infer.py -------------------------------------------------------------------------------- /KnowledgeDistillation/KnowledgeDistillation/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/KnowledgeDistillation/infer.sh -------------------------------------------------------------------------------- /KnowledgeDistillation/KnowledgeDistillation/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/KnowledgeDistillation/model.py -------------------------------------------------------------------------------- /KnowledgeDistillation/KnowledgeDistillation/requirements.txt: -------------------------------------------------------------------------------- 1 | flowvision==0.2.0 2 | matplotlib==3.4.3 3 | numpy==1.21.2 4 | -------------------------------------------------------------------------------- /KnowledgeDistillation/KnowledgeDistillation/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/KnowledgeDistillation/train.py -------------------------------------------------------------------------------- /KnowledgeDistillation/KnowledgeDistillation/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/KnowledgeDistillation/train.sh -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/1_get_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/1_get_data.sh -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/2_process_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/2_process_data.sh -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/3_train_teacher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/3_train_teacher.sh -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/4_student_first.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/4_student_first.sh -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/5_student_second.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/5_student_second.sh -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/6_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/6_eval.sh -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/README.md -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/bert.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/bert_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/bert_utils.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/appzoo/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/appzoo/api.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/appzoo/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/appzoo/dataset.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/core/__init__.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/core/distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/core/distiller.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/core/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/core/evaluator.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/core/trainer.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/requirements.txt -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/utils/__init__.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/utils/exporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/utils/exporter.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/utils/io_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/utils/io_utils.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/utils/logger.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/easynlp/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/easynlp/utils/losses.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/extract_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/extract_embeddings.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/generate_dev_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/generate_dev_file.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/generate_meta_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/generate_meta_weights.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/generate_senti_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/generate_senti_data.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/main_evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/main_evaluate.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/meta_student_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/meta_student_distill.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/meta_teacher_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/meta_teacher_train.py -------------------------------------------------------------------------------- /KnowledgeDistillation/MetaKD/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/KnowledgeDistillation/MetaKD/requirements.txt -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/LICENSE -------------------------------------------------------------------------------- /NLP/BERT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/README.md -------------------------------------------------------------------------------- /NLP/BERT/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/config.py -------------------------------------------------------------------------------- /NLP/BERT/examples/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/examples/infer.sh -------------------------------------------------------------------------------- /NLP/BERT/examples/run_eager_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/examples/run_eager_pretrain.sh -------------------------------------------------------------------------------- /NLP/BERT/examples/run_graph_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/examples/run_graph_pretrain.sh -------------------------------------------------------------------------------- /NLP/BERT/examples/run_squad.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/examples/run_squad.sh -------------------------------------------------------------------------------- /NLP/BERT/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/modeling.py -------------------------------------------------------------------------------- /NLP/BERT/run_eager_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/run_eager_pretraining.py -------------------------------------------------------------------------------- /NLP/BERT/run_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/run_infer.py -------------------------------------------------------------------------------- /NLP/BERT/run_pretraining.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/run_pretraining.py -------------------------------------------------------------------------------- /NLP/BERT/run_squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/run_squad.py -------------------------------------------------------------------------------- /NLP/BERT/squad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/squad.py -------------------------------------------------------------------------------- /NLP/BERT/squad_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/squad_util.py -------------------------------------------------------------------------------- /NLP/BERT/tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/tokenization.py -------------------------------------------------------------------------------- /NLP/BERT/utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/checkpoint.py -------------------------------------------------------------------------------- /NLP/BERT/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/comm.py -------------------------------------------------------------------------------- /NLP/BERT/utils/compare_lazy_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/compare_lazy_outputs.py -------------------------------------------------------------------------------- /NLP/BERT/utils/draw_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/draw_img.py -------------------------------------------------------------------------------- /NLP/BERT/utils/draw_loss_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/draw_loss_curve.py -------------------------------------------------------------------------------- /NLP/BERT/utils/lamb_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/lamb_optimizer.py -------------------------------------------------------------------------------- /NLP/BERT/utils/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/lr_scheduler.py -------------------------------------------------------------------------------- /NLP/BERT/utils/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/metric.py -------------------------------------------------------------------------------- /NLP/BERT/utils/ofrecord_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/ofrecord_data_utils.py -------------------------------------------------------------------------------- /NLP/BERT/utils/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/optimizer.py -------------------------------------------------------------------------------- /NLP/BERT/utils/reporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/BERT/utils/reporter.py -------------------------------------------------------------------------------- /NLP/CPT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/README.md -------------------------------------------------------------------------------- /NLP/CPT/classifier_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/classifier_flow.py -------------------------------------------------------------------------------- /NLP/CPT/dataset_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/dataset_flow.py -------------------------------------------------------------------------------- /NLP/CPT/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/infer.sh -------------------------------------------------------------------------------- /NLP/CPT/infer_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/infer_flow.py -------------------------------------------------------------------------------- /NLP/CPT/models/CPT.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/models/CPT.py -------------------------------------------------------------------------------- /NLP/CPT/models/bart_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/models/bart_utils.py -------------------------------------------------------------------------------- /NLP/CPT/models/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/models/bert.py -------------------------------------------------------------------------------- /NLP/CPT/models/bert_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/models/bert_utils.py -------------------------------------------------------------------------------- /NLP/CPT/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/models/utils.py -------------------------------------------------------------------------------- /NLP/CPT/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/requirements.txt -------------------------------------------------------------------------------- /NLP/CPT/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/train.sh -------------------------------------------------------------------------------- /NLP/CPT/train_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/CPT/train_flow.py -------------------------------------------------------------------------------- /NLP/GPT2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/README.md -------------------------------------------------------------------------------- /NLP/GPT2/compare_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/compare_speed.py -------------------------------------------------------------------------------- /NLP/GPT2/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/config.py -------------------------------------------------------------------------------- /NLP/GPT2/convert_pt_ckpt_to_of.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/convert_pt_ckpt_to_of.py -------------------------------------------------------------------------------- /NLP/GPT2/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/finetune.sh -------------------------------------------------------------------------------- /NLP/GPT2/generate_sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/generate_sample.py -------------------------------------------------------------------------------- /NLP/GPT2/gpt_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/gpt_dataset.py -------------------------------------------------------------------------------- /NLP/GPT2/gpt_dataset_pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/gpt_dataset_pt.py -------------------------------------------------------------------------------- /NLP/GPT2/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/infer.sh -------------------------------------------------------------------------------- /NLP/GPT2/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/model.py -------------------------------------------------------------------------------- /NLP/GPT2/model_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/model_config.py -------------------------------------------------------------------------------- /NLP/GPT2/optimization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/optimization.py -------------------------------------------------------------------------------- /NLP/GPT2/optimization_pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/optimization_pt.py -------------------------------------------------------------------------------- /NLP/GPT2/pt_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/pt_model.py -------------------------------------------------------------------------------- /NLP/GPT2/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/requirements.txt -------------------------------------------------------------------------------- /NLP/GPT2/run_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/run_generation.py -------------------------------------------------------------------------------- /NLP/GPT2/tokenizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/tokenizer/__init__.py -------------------------------------------------------------------------------- /NLP/GPT2/tokenizer/gpt2_tokenization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/tokenizer/gpt2_tokenization.py -------------------------------------------------------------------------------- /NLP/GPT2/tokenizer/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/tokenizer/tokenizer.py -------------------------------------------------------------------------------- /NLP/GPT2/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/train.py -------------------------------------------------------------------------------- /NLP/GPT2/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/train.sh -------------------------------------------------------------------------------- /NLP/GPT2/train_pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/train_pt.py -------------------------------------------------------------------------------- /NLP/GPT2/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/trainer.py -------------------------------------------------------------------------------- /NLP/GPT2/trainer_pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/GPT2/trainer_pt.py -------------------------------------------------------------------------------- /NLP/LSTMText/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/LSTMText/README.md -------------------------------------------------------------------------------- /NLP/LSTMText/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/LSTMText/infer.py -------------------------------------------------------------------------------- /NLP/LSTMText/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/LSTMText/infer.sh -------------------------------------------------------------------------------- /NLP/LSTMText/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/LSTMText/model.py -------------------------------------------------------------------------------- /NLP/LSTMText/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | six 3 | shutil -------------------------------------------------------------------------------- /NLP/LSTMText/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/LSTMText/train.sh -------------------------------------------------------------------------------- /NLP/LSTMText/train_bilstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/LSTMText/train_bilstm.py -------------------------------------------------------------------------------- /NLP/LSTMText/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/LSTMText/utils.py -------------------------------------------------------------------------------- /NLP/MoE/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/MoE/README.md -------------------------------------------------------------------------------- /NLP/MoE/cifar10_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/MoE/cifar10_example.py -------------------------------------------------------------------------------- /NLP/MoE/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/MoE/mlp.py -------------------------------------------------------------------------------- /NLP/MoE/moe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/MoE/moe.py -------------------------------------------------------------------------------- /NLP/MoE/requirements.txt: -------------------------------------------------------------------------------- 1 | flowvision==0.0.3 2 | -------------------------------------------------------------------------------- /NLP/MoE/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/MoE/train.sh -------------------------------------------------------------------------------- /NLP/TextCNN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/TextCNN/README.md -------------------------------------------------------------------------------- /NLP/TextCNN/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/TextCNN/infer.py -------------------------------------------------------------------------------- /NLP/TextCNN/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/TextCNN/infer.sh -------------------------------------------------------------------------------- /NLP/TextCNN/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/TextCNN/model.py -------------------------------------------------------------------------------- /NLP/TextCNN/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/TextCNN/requirements.txt -------------------------------------------------------------------------------- /NLP/TextCNN/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/TextCNN/train.sh -------------------------------------------------------------------------------- /NLP/TextCNN/train_of_textcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/TextCNN/train_of_textcnn.py -------------------------------------------------------------------------------- /NLP/TextCNN/training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/TextCNN/training.py -------------------------------------------------------------------------------- /NLP/TextCNN/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/TextCNN/utils.py -------------------------------------------------------------------------------- /NLP/Transformer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/README.md -------------------------------------------------------------------------------- /NLP/Transformer/imdb/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/imdb/infer.sh -------------------------------------------------------------------------------- /NLP/Transformer/imdb/infer_transformer_imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/imdb/infer_transformer_imdb.py -------------------------------------------------------------------------------- /NLP/Transformer/imdb/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/imdb/model.py -------------------------------------------------------------------------------- /NLP/Transformer/imdb/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/imdb/train.sh -------------------------------------------------------------------------------- /NLP/Transformer/imdb/train_transformer_imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/imdb/train_transformer_imdb.py -------------------------------------------------------------------------------- /NLP/Transformer/imdb/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/imdb/utils.py -------------------------------------------------------------------------------- /NLP/Transformer/odd_numbers/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/odd_numbers/infer.sh -------------------------------------------------------------------------------- /NLP/Transformer/odd_numbers/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/odd_numbers/model.py -------------------------------------------------------------------------------- /NLP/Transformer/odd_numbers/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/odd_numbers/train.sh -------------------------------------------------------------------------------- /NLP/Transformer/transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/transformer/__init__.py -------------------------------------------------------------------------------- /NLP/Transformer/transformer/multihead_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/transformer/multihead_attention.py -------------------------------------------------------------------------------- /NLP/Transformer/transformer/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/transformer/transformer.py -------------------------------------------------------------------------------- /NLP/Transformer/transformer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/Transformer/transformer/utils.py -------------------------------------------------------------------------------- /NLP/rnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/README.md -------------------------------------------------------------------------------- /NLP/rnn/compare_oneflow_and_pytorch_lstm_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/compare_oneflow_and_pytorch_lstm_speed.py -------------------------------------------------------------------------------- /NLP/rnn/compare_oneflow_and_pytorch_rnn_speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/compare_oneflow_and_pytorch_rnn_speed.py -------------------------------------------------------------------------------- /NLP/rnn/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /NLP/rnn/models/lstm_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/models/lstm_oneflow.py -------------------------------------------------------------------------------- /NLP/rnn/models/lstm_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/models/lstm_pytorch.py -------------------------------------------------------------------------------- /NLP/rnn/models/rnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/models/rnn_model.py -------------------------------------------------------------------------------- /NLP/rnn/models/rnn_model_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/models/rnn_model_pytorch.py -------------------------------------------------------------------------------- /NLP/rnn/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/predict.py -------------------------------------------------------------------------------- /NLP/rnn/requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | numpy 3 | -------------------------------------------------------------------------------- /NLP/rnn/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/train.sh -------------------------------------------------------------------------------- /NLP/rnn/train_lstm_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/train_lstm_oneflow.py -------------------------------------------------------------------------------- /NLP/rnn/train_rnn_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/train_rnn_oneflow.py -------------------------------------------------------------------------------- /NLP/rnn/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /NLP/rnn/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/utils/dataset.py -------------------------------------------------------------------------------- /NLP/rnn/utils/tensor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/rnn/utils/tensor_utils.py -------------------------------------------------------------------------------- /NLP/roberta/MNLIDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/MNLIDataset.py -------------------------------------------------------------------------------- /NLP/roberta/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/README.md -------------------------------------------------------------------------------- /NLP/roberta/SST2Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/SST2Dataset.py -------------------------------------------------------------------------------- /NLP/roberta/classifier_MNLI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/classifier_MNLI.py -------------------------------------------------------------------------------- /NLP/roberta/classifier_SST2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/classifier_SST2.py -------------------------------------------------------------------------------- /NLP/roberta/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/config.py -------------------------------------------------------------------------------- /NLP/roberta/infer_MNLI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/infer_MNLI.py -------------------------------------------------------------------------------- /NLP/roberta/infer_MNLI.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/infer_MNLI.sh -------------------------------------------------------------------------------- /NLP/roberta/infer_SST2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/infer_SST2.py -------------------------------------------------------------------------------- /NLP/roberta/infer_SST2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/infer_SST2.sh -------------------------------------------------------------------------------- /NLP/roberta/models/roberta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/models/roberta.py -------------------------------------------------------------------------------- /NLP/roberta/models/roberta_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/models/roberta_utils.py -------------------------------------------------------------------------------- /NLP/roberta/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/requirements.txt -------------------------------------------------------------------------------- /NLP/roberta/train_MNLI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/train_MNLI.py -------------------------------------------------------------------------------- /NLP/roberta/train_MNLI.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/train_MNLI.sh -------------------------------------------------------------------------------- /NLP/roberta/train_SST2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/train_SST2.py -------------------------------------------------------------------------------- /NLP/roberta/train_SST2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/roberta/train_SST2.sh -------------------------------------------------------------------------------- /NLP/seq2seq/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/seq2seq/README.md -------------------------------------------------------------------------------- /NLP/seq2seq/eval_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/seq2seq/eval_oneflow.py -------------------------------------------------------------------------------- /NLP/seq2seq/eval_oneflow.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/seq2seq/eval_oneflow.sh -------------------------------------------------------------------------------- /NLP/seq2seq/models/GRU_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/seq2seq/models/GRU_oneflow.py -------------------------------------------------------------------------------- /NLP/seq2seq/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /NLP/seq2seq/models/seq_seq_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/seq2seq/models/seq_seq_oneflow.py -------------------------------------------------------------------------------- /NLP/seq2seq/requirements.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | numpy -------------------------------------------------------------------------------- /NLP/seq2seq/train_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/seq2seq/train_oneflow.py -------------------------------------------------------------------------------- /NLP/seq2seq/train_oneflow.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/seq2seq/train_oneflow.sh -------------------------------------------------------------------------------- /NLP/seq2seq/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /NLP/seq2seq/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/seq2seq/utils/dataset.py -------------------------------------------------------------------------------- /NLP/seq2seq/utils/utils_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/NLP/seq2seq/utils/utils_oneflow.py -------------------------------------------------------------------------------- /Quantization/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/README.md -------------------------------------------------------------------------------- /Quantization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Quantization/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/infer.sh -------------------------------------------------------------------------------- /Quantization/models/q_alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/models/q_alexnet.py -------------------------------------------------------------------------------- /Quantization/quantization_aware_training.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/quantization_aware_training.py -------------------------------------------------------------------------------- /Quantization/quantization_infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/quantization_infer.py -------------------------------------------------------------------------------- /Quantization/quantization_ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/quantization_ops/__init__.py -------------------------------------------------------------------------------- /Quantization/quantization_ops/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/quantization_ops/conv.py -------------------------------------------------------------------------------- /Quantization/quantization_ops/conv_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/quantization_ops/conv_bn.py -------------------------------------------------------------------------------- /Quantization/quantization_ops/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/quantization_ops/linear.py -------------------------------------------------------------------------------- /Quantization/quantization_ops/q_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/quantization_ops/q_module.py -------------------------------------------------------------------------------- /Quantization/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/train.sh -------------------------------------------------------------------------------- /Quantization/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Quantization/utils/imagenet1000_clsidx_to_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/utils/imagenet1000_clsidx_to_labels.py -------------------------------------------------------------------------------- /Quantization/utils/numpy_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/utils/numpy_data_utils.py -------------------------------------------------------------------------------- /Quantization/utils/ofrecord_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/utils/ofrecord_data_utils.py -------------------------------------------------------------------------------- /Quantization/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Quantization/utils/plot.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/README.md -------------------------------------------------------------------------------- /README_zh-CN.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/README_zh-CN.md -------------------------------------------------------------------------------- /RecommenderSystems/dcn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dcn/README.md -------------------------------------------------------------------------------- /RecommenderSystems/dcn/dcn_eager_train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dcn/dcn_eager_train_eval.py -------------------------------------------------------------------------------- /RecommenderSystems/dcn/dcn_train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dcn/dcn_train_eval.py -------------------------------------------------------------------------------- /RecommenderSystems/dcn/eager_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dcn/eager_train.sh -------------------------------------------------------------------------------- /RecommenderSystems/dcn/requirements.txt: -------------------------------------------------------------------------------- 1 | petastorm 2 | psutil 3 | pandas 4 | sklearn 5 | 6 | 7 | -------------------------------------------------------------------------------- /RecommenderSystems/dcn/tools/dcn_parquet.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dcn/tools/dcn_parquet.scala -------------------------------------------------------------------------------- /RecommenderSystems/dcn/tools/launch_spark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dcn/tools/launch_spark.sh -------------------------------------------------------------------------------- /RecommenderSystems/dcn/tools/split_criteo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dcn/tools/split_criteo.py -------------------------------------------------------------------------------- /RecommenderSystems/dcn/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dcn/train.sh -------------------------------------------------------------------------------- /RecommenderSystems/deepfm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/deepfm/README.md -------------------------------------------------------------------------------- /RecommenderSystems/deepfm/deepfm_eager_train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/deepfm/deepfm_eager_train_eval.py -------------------------------------------------------------------------------- /RecommenderSystems/deepfm/deepfm_train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/deepfm/deepfm_train_eval.py -------------------------------------------------------------------------------- /RecommenderSystems/deepfm/eager_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/deepfm/eager_train.sh -------------------------------------------------------------------------------- /RecommenderSystems/deepfm/tools/deepfm_parquet.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/deepfm/tools/deepfm_parquet.scala -------------------------------------------------------------------------------- /RecommenderSystems/deepfm/tools/launch_spark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/deepfm/tools/launch_spark.sh -------------------------------------------------------------------------------- /RecommenderSystems/deepfm/tools/split_criteo_kaggle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/deepfm/tools/split_criteo_kaggle.py -------------------------------------------------------------------------------- /RecommenderSystems/deepfm/train_deepfm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/deepfm/train_deepfm.sh -------------------------------------------------------------------------------- /RecommenderSystems/dlrm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dlrm/README.md -------------------------------------------------------------------------------- /RecommenderSystems/dlrm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /RecommenderSystems/dlrm/dlrm_train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dlrm/dlrm_train_eval.py -------------------------------------------------------------------------------- /RecommenderSystems/dlrm/requirements.txt: -------------------------------------------------------------------------------- 1 | petastorm 2 | psutil 3 | -------------------------------------------------------------------------------- /RecommenderSystems/dlrm/tools/criteo1t_parquet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dlrm/tools/criteo1t_parquet.py -------------------------------------------------------------------------------- /RecommenderSystems/dlrm/tools/criteo1t_parquet.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dlrm/tools/criteo1t_parquet.scala -------------------------------------------------------------------------------- /RecommenderSystems/dlrm/tools/launch_spark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dlrm/tools/launch_spark.sh -------------------------------------------------------------------------------- /RecommenderSystems/dlrm/tools/split_day_23.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/dlrm/tools/split_day_23.sh -------------------------------------------------------------------------------- /RecommenderSystems/mmoe/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/mmoe/README.md -------------------------------------------------------------------------------- /RecommenderSystems/mmoe/mmoe_train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/mmoe/mmoe_train_eval.py -------------------------------------------------------------------------------- /RecommenderSystems/mmoe/tools/mmoe_parquet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/mmoe/tools/mmoe_parquet.py -------------------------------------------------------------------------------- /RecommenderSystems/mmoe/train_mmoe.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/mmoe/train_mmoe.sh -------------------------------------------------------------------------------- /RecommenderSystems/pnn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/pnn/README.md -------------------------------------------------------------------------------- /RecommenderSystems/pnn/pnn_train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/pnn/pnn_train_eval.py -------------------------------------------------------------------------------- /RecommenderSystems/pnn/tools/criteo_parquet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/pnn/tools/criteo_parquet.py -------------------------------------------------------------------------------- /RecommenderSystems/pnn/tools/launch_spark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/pnn/tools/launch_spark.sh -------------------------------------------------------------------------------- /RecommenderSystems/pnn/tools/split_criteo_kaggle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/pnn/tools/split_criteo_kaggle.py -------------------------------------------------------------------------------- /RecommenderSystems/pnn/train_pnn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/pnn/train_pnn.sh -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/wide_and_deep/README.md -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/wide_and_deep/config.py -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/wide_and_deep/graph.py -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/models/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/wide_and_deep/models/data.py -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/wide_and_deep/train.py -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/train_ddp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/wide_and_deep/train_ddp.sh -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/train_global_eager.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/wide_and_deep/train_global_eager.sh -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/train_global_graph.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/wide_and_deep/train_global_graph.sh -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/train_nn_graph.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/wide_and_deep/train_nn_graph.sh -------------------------------------------------------------------------------- /RecommenderSystems/wide_and_deep/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/wide_and_deep/utils/logger.py -------------------------------------------------------------------------------- /RecommenderSystems/xdeepfm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/xdeepfm/README.md -------------------------------------------------------------------------------- /RecommenderSystems/xdeepfm/tools/launch_spark.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/xdeepfm/tools/launch_spark.sh -------------------------------------------------------------------------------- /RecommenderSystems/xdeepfm/tools/xdeepfm_parquet.scala: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/xdeepfm/tools/xdeepfm_parquet.scala -------------------------------------------------------------------------------- /RecommenderSystems/xdeepfm/train_xdeepfm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/xdeepfm/train_xdeepfm.sh -------------------------------------------------------------------------------- /RecommenderSystems/xdeepfm/xdeepfm_train_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/RecommenderSystems/xdeepfm/xdeepfm_train_eval.py -------------------------------------------------------------------------------- /Vision/BJTU_summer_tournament/DataSet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/BJTU_summer_tournament/DataSet.py -------------------------------------------------------------------------------- /Vision/BJTU_summer_tournament/DenseNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/BJTU_summer_tournament/DenseNet.py -------------------------------------------------------------------------------- /Vision/BJTU_summer_tournament/Predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/BJTU_summer_tournament/Predict.py -------------------------------------------------------------------------------- /Vision/BJTU_summer_tournament/Prepare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/BJTU_summer_tournament/Prepare.py -------------------------------------------------------------------------------- /Vision/BJTU_summer_tournament/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/BJTU_summer_tournament/README.md -------------------------------------------------------------------------------- /Vision/BJTU_summer_tournament/Train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/BJTU_summer_tournament/Train.py -------------------------------------------------------------------------------- /Vision/BJTU_summer_tournament/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/BJTU_summer_tournament/requirements.txt -------------------------------------------------------------------------------- /Vision/LSTM/EnResLSTM/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/LSTM/EnResLSTM/README.md -------------------------------------------------------------------------------- /Vision/LSTM/EnResLSTM/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/LSTM/EnResLSTM/infer.py -------------------------------------------------------------------------------- /Vision/LSTM/EnResLSTM/run.sh: -------------------------------------------------------------------------------- 1 | python train.py -------------------------------------------------------------------------------- /Vision/LSTM/EnResLSTM/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/LSTM/EnResLSTM/train.py -------------------------------------------------------------------------------- /Vision/classification/image/DLA/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/DLA/README.md -------------------------------------------------------------------------------- /Vision/classification/image/DLA/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/DLA/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/DLA/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/DLA/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/DLA/models/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/DLA/models/dla.py -------------------------------------------------------------------------------- /Vision/classification/image/DLA/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy==1.19.5 2 | Pillow==8.3.1 3 | -------------------------------------------------------------------------------- /Vision/classification/image/DLA/train_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/DLA/train_oneflow.py -------------------------------------------------------------------------------- /Vision/classification/image/DLA/train_oneflow.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/DLA/train_oneflow.sh -------------------------------------------------------------------------------- /Vision/classification/image/ViT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/README.md -------------------------------------------------------------------------------- /Vision/classification/image/ViT/eager/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/eager/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/ViT/eager/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/eager/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/ViT/eager/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/eager/train.py -------------------------------------------------------------------------------- /Vision/classification/image/ViT/eager/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/eager/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/ViT/graph/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/graph/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/ViT/graph/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/graph/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/ViT/graph/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/graph/train.py -------------------------------------------------------------------------------- /Vision/classification/image/ViT/graph/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/graph/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/ViT/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/ViT/model/build_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/model/build_model.py -------------------------------------------------------------------------------- /Vision/classification/image/ViT/model/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ViT/model/vit.py -------------------------------------------------------------------------------- /Vision/classification/image/ViT/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | Pillow 4 | -------------------------------------------------------------------------------- /Vision/classification/image/ViT/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/README.md -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/check/check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/check/check.py -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/check/check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/check/check.sh -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/check/draw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/check/draw.py -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/check/draw.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/check/draw.sh -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/eager/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/eager/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/eager/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/eager/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/eager/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/eager/train.py -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/eager/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/eager/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/graph/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/graph/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/graph/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/graph/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/graph/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/graph/train.py -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/graph/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/graph/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/model/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/model/alexnet.py -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | Pillow 4 | tqdm -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/alexnet/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/alexnet/utils/plot.py -------------------------------------------------------------------------------- /Vision/classification/image/densenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/densenet/README.md -------------------------------------------------------------------------------- /Vision/classification/image/densenet/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/densenet/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/densenet/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/densenet/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/densenet/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/densenet/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | Pillow 4 | -------------------------------------------------------------------------------- /Vision/classification/image/densenet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/densenet/train.py -------------------------------------------------------------------------------- /Vision/classification/image/densenet/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/densenet/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/densenet/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/densenet/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/densenet/utils/plot.py -------------------------------------------------------------------------------- /Vision/classification/image/ghostnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ghostnet/README.md -------------------------------------------------------------------------------- /Vision/classification/image/ghostnet/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ghostnet/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/ghostnet/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ghostnet/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/ghostnet/train_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ghostnet/train_oneflow.py -------------------------------------------------------------------------------- /Vision/classification/image/ghostnet/train_oneflow.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/ghostnet/train_oneflow.sh -------------------------------------------------------------------------------- /Vision/classification/image/inception_v3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/inception_v3/README.md -------------------------------------------------------------------------------- /Vision/classification/image/inception_v3/graph/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/inception_v3/graph/data.py -------------------------------------------------------------------------------- /Vision/classification/image/inception_v3/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/inception_v3/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | pillow 4 | scipy 5 | -------------------------------------------------------------------------------- /Vision/classification/image/inception_v3/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/inception_v3/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/inception_v3/utils/plot.py -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/README.md -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/eager/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/eager/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/eager/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/eager/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/eager/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/eager/train.py -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/eager/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/eager/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/graph/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/graph/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/graph/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/graph/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/graph/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/graph/train.py -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/graph/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/graph/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/model/mnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/model/mnasnet.py -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/mnasnet/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mnasnet/utils/plot.py -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv2/README.md -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv2/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv2/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv2/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv2/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv2/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv2/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | Pillow 4 | -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv2/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv2/train.py -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv2/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv2/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv2/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv2/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv2/utils/plot.py -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv3/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv3/README.md -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv3/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv3/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv3/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv3/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv3/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv3/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | Pillow 4 | -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv3/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv3/train.py -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv3/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv3/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv3/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/mobilenetv3/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/mobilenetv3/utils/plot.py -------------------------------------------------------------------------------- /Vision/classification/image/poseNet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/poseNet/README.md -------------------------------------------------------------------------------- /Vision/classification/image/poseNet/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/poseNet/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/poseNet/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/poseNet/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/poseNet/models/posenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/poseNet/models/posenet.py -------------------------------------------------------------------------------- /Vision/classification/image/poseNet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/poseNet/requirements.txt -------------------------------------------------------------------------------- /Vision/classification/image/poseNet/train_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/poseNet/train_oneflow.py -------------------------------------------------------------------------------- /Vision/classification/image/poseNet/train_oneflow.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/poseNet/train_oneflow.sh -------------------------------------------------------------------------------- /Vision/classification/image/repvgg/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/repvgg/README.md -------------------------------------------------------------------------------- /Vision/classification/image/repvgg/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/repvgg/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/repvgg/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/repvgg/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/repvgg/models/repvgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/repvgg/models/repvgg.py -------------------------------------------------------------------------------- /Vision/classification/image/repvgg/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | Pillow 4 | -------------------------------------------------------------------------------- /Vision/classification/image/repvgg/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/repvgg/train.py -------------------------------------------------------------------------------- /Vision/classification/image/repvgg/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/repvgg/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/repvgg/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/repvgg/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/repvgg/utils/plot.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/README.md -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/check/check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/check/check.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/check/check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/check/check.sh -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/check/draw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/check/draw.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/check/draw.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/check/draw.sh -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/config.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/examples/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/examples/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/graph.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/models/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/models/data.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | torch 4 | Pillow 5 | -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/train.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/utils/debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/utils/debug.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/utils/logger.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/utils/plot.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/utils/printer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/utils/printer.py -------------------------------------------------------------------------------- /Vision/classification/image/resnet50/utils/stat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnet50/utils/stat.py -------------------------------------------------------------------------------- /Vision/classification/image/resnext50_32x4d/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnext50_32x4d/README.md -------------------------------------------------------------------------------- /Vision/classification/image/resnext50_32x4d/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnext50_32x4d/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/resnext50_32x4d/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnext50_32x4d/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/resnext50_32x4d/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/resnext50_32x4d/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | Pillow 4 | -------------------------------------------------------------------------------- /Vision/classification/image/resnext50_32x4d/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnext50_32x4d/train.py -------------------------------------------------------------------------------- /Vision/classification/image/resnext50_32x4d/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/resnext50_32x4d/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/resnext50_32x4d/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/scnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/scnet/README.md -------------------------------------------------------------------------------- /Vision/classification/image/scnet/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/scnet/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/scnet/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/scnet/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/scnet/models/resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/scnet/models/resnet50.py -------------------------------------------------------------------------------- /Vision/classification/image/scnet/models/scloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/scnet/models/scloss.py -------------------------------------------------------------------------------- /Vision/classification/image/scnet/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | Pillow -------------------------------------------------------------------------------- /Vision/classification/image/scnet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/scnet/train.py -------------------------------------------------------------------------------- /Vision/classification/image/scnet/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/scnet/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/shufflenetv2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/shufflenetv2/README.md -------------------------------------------------------------------------------- /Vision/classification/image/shufflenetv2/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/shufflenetv2/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/shufflenetv2/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/shufflenetv2/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/shufflenetv2/requirements.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | numpy 3 | Pillow 4 | -------------------------------------------------------------------------------- /Vision/classification/image/shufflenetv2/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/shufflenetv2/train.py -------------------------------------------------------------------------------- /Vision/classification/image/shufflenetv2/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/shufflenetv2/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/shufflenetv2/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/shufflenetv2/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/shufflenetv2/utils/plot.py -------------------------------------------------------------------------------- /Vision/classification/image/swin_transformer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/swin_transformer/README.md -------------------------------------------------------------------------------- /Vision/classification/image/swin_transformer/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/swin_transformer/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/swin_transformer/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/swin_transformer/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/swin_transformer/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | matplotlib 3 | Pillow 4 | -------------------------------------------------------------------------------- /Vision/classification/image/swin_transformer/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/swin_transformer/train.py -------------------------------------------------------------------------------- /Vision/classification/image/swin_transformer/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/swin_transformer/train.sh -------------------------------------------------------------------------------- /Vision/classification/image/swin_transformer/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/vgg/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/vgg/README.md -------------------------------------------------------------------------------- /Vision/classification/image/vgg/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/vgg/infer.py -------------------------------------------------------------------------------- /Vision/classification/image/vgg/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/vgg/infer.sh -------------------------------------------------------------------------------- /Vision/classification/image/vgg/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/vgg/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/vgg/models/vgg.py -------------------------------------------------------------------------------- /Vision/classification/image/vgg/train_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/vgg/train_oneflow.py -------------------------------------------------------------------------------- /Vision/classification/image/vgg/train_oneflow.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/vgg/train_oneflow.sh -------------------------------------------------------------------------------- /Vision/classification/image/vgg/utils/Figure_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/vgg/utils/Figure_1.png -------------------------------------------------------------------------------- /Vision/classification/image/vgg/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/classification/image/vgg/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/image/vgg/utils/plot.py -------------------------------------------------------------------------------- /Vision/classification/video/TSN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/video/TSN/README.md -------------------------------------------------------------------------------- /Vision/classification/video/TSN/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/video/TSN/infer.sh -------------------------------------------------------------------------------- /Vision/classification/video/TSN/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | PIL 3 | -------------------------------------------------------------------------------- /Vision/classification/video/TSN/test_recognizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/video/TSN/test_recognizer.py -------------------------------------------------------------------------------- /Vision/classification/video/TSN/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/video/TSN/train.sh -------------------------------------------------------------------------------- /Vision/classification/video/TSN/train_recognizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/video/TSN/train_recognizer.py -------------------------------------------------------------------------------- /Vision/classification/video/TSN/tsn/models/TSN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/video/TSN/tsn/models/TSN.py -------------------------------------------------------------------------------- /Vision/classification/video/TSN/tsn/models/cls_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/video/TSN/tsn/models/cls_head.py -------------------------------------------------------------------------------- /Vision/classification/video/TSN/tsn/models/resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/classification/video/TSN/tsn/models/resnet50.py -------------------------------------------------------------------------------- /Vision/detection/CSRNet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/README.md -------------------------------------------------------------------------------- /Vision/detection/CSRNet/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/dataset.py -------------------------------------------------------------------------------- /Vision/detection/CSRNet/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/image.py -------------------------------------------------------------------------------- /Vision/detection/CSRNet/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/infer.py -------------------------------------------------------------------------------- /Vision/detection/CSRNet/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/infer.sh -------------------------------------------------------------------------------- /Vision/detection/CSRNet/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/model.py -------------------------------------------------------------------------------- /Vision/detection/CSRNet/part_A_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/part_A_test.json -------------------------------------------------------------------------------- /Vision/detection/CSRNet/part_A_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/part_A_train.json -------------------------------------------------------------------------------- /Vision/detection/CSRNet/part_A_train_with_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/part_A_train_with_val.json -------------------------------------------------------------------------------- /Vision/detection/CSRNet/part_A_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/part_A_val.json -------------------------------------------------------------------------------- /Vision/detection/CSRNet/part_B_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/part_B_test.json -------------------------------------------------------------------------------- /Vision/detection/CSRNet/part_B_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/part_B_train.json -------------------------------------------------------------------------------- /Vision/detection/CSRNet/part_B_train_with_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/part_B_train_with_val.json -------------------------------------------------------------------------------- /Vision/detection/CSRNet/part_B_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/part_B_val.json -------------------------------------------------------------------------------- /Vision/detection/CSRNet/requirement.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | cv2 3 | h5py 4 | pillow -------------------------------------------------------------------------------- /Vision/detection/CSRNet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/train.py -------------------------------------------------------------------------------- /Vision/detection/CSRNet/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/train.sh -------------------------------------------------------------------------------- /Vision/detection/CSRNet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/utils.py -------------------------------------------------------------------------------- /Vision/detection/CSRNet/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/val.py -------------------------------------------------------------------------------- /Vision/detection/CSRNet/val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/val.sh -------------------------------------------------------------------------------- /Vision/detection/CSRNet/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/detection/CSRNet/vgg.py -------------------------------------------------------------------------------- /Vision/gan/DCGAN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/README.md -------------------------------------------------------------------------------- /Vision/gan/DCGAN/check/check.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/check/check.py -------------------------------------------------------------------------------- /Vision/gan/DCGAN/check/check.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/check/check.sh -------------------------------------------------------------------------------- /Vision/gan/DCGAN/eager/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/eager/train.py -------------------------------------------------------------------------------- /Vision/gan/DCGAN/eager/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/eager/train.sh -------------------------------------------------------------------------------- /Vision/gan/DCGAN/graph/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/graph/train.py -------------------------------------------------------------------------------- /Vision/gan/DCGAN/graph/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/graph/train.sh -------------------------------------------------------------------------------- /Vision/gan/DCGAN/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/models.py -------------------------------------------------------------------------------- /Vision/gan/DCGAN/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/plot.py -------------------------------------------------------------------------------- /Vision/gan/DCGAN/test/test_of_dcgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/test/test_of_dcgan.py -------------------------------------------------------------------------------- /Vision/gan/DCGAN/test/test_of_dcgan.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/test/test_of_dcgan.sh -------------------------------------------------------------------------------- /Vision/gan/DCGAN/test/test_of_dcgan_graph.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/test/test_of_dcgan_graph.sh -------------------------------------------------------------------------------- /Vision/gan/DCGAN/test_images.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/test_images.png -------------------------------------------------------------------------------- /Vision/gan/DCGAN/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/DCGAN/utils.py -------------------------------------------------------------------------------- /Vision/gan/SRGAN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/README.md -------------------------------------------------------------------------------- /Vision/gan/SRGAN/models/of_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/models/of_model.py -------------------------------------------------------------------------------- /Vision/gan/SRGAN/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/requirements.txt -------------------------------------------------------------------------------- /Vision/gan/SRGAN/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/results.png -------------------------------------------------------------------------------- /Vision/gan/SRGAN/test_of_srgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/test_of_srgan.py -------------------------------------------------------------------------------- /Vision/gan/SRGAN/test_of_srgan.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/test_of_srgan.sh -------------------------------------------------------------------------------- /Vision/gan/SRGAN/train_of_srgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/train_of_srgan.py -------------------------------------------------------------------------------- /Vision/gan/SRGAN/train_of_srgan.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/train_of_srgan.sh -------------------------------------------------------------------------------- /Vision/gan/SRGAN/utils/of_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/utils/of_data_utils.py -------------------------------------------------------------------------------- /Vision/gan/SRGAN/utils/of_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/utils/of_loss.py -------------------------------------------------------------------------------- /Vision/gan/SRGAN/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/SRGAN/vgg.py -------------------------------------------------------------------------------- /Vision/gan/cycleGAN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/cycleGAN/README.md -------------------------------------------------------------------------------- /Vision/gan/cycleGAN/cycleGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/cycleGAN/cycleGAN.py -------------------------------------------------------------------------------- /Vision/gan/cycleGAN/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/cycleGAN/download.sh -------------------------------------------------------------------------------- /Vision/gan/cycleGAN/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/cycleGAN/image.py -------------------------------------------------------------------------------- /Vision/gan/cycleGAN/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/cycleGAN/infer.py -------------------------------------------------------------------------------- /Vision/gan/cycleGAN/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/cycleGAN/infer.sh -------------------------------------------------------------------------------- /Vision/gan/cycleGAN/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/cycleGAN/networks.py -------------------------------------------------------------------------------- /Vision/gan/cycleGAN/requirements.txt: -------------------------------------------------------------------------------- 1 | opencv_python==4.2.0.34 2 | numpy==1.20.1 3 | -------------------------------------------------------------------------------- /Vision/gan/cycleGAN/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/cycleGAN/train.py -------------------------------------------------------------------------------- /Vision/gan/cycleGAN/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/cycleGAN/train.sh -------------------------------------------------------------------------------- /Vision/gan/pix2pix/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/pix2pix/README.md -------------------------------------------------------------------------------- /Vision/gan/pix2pix/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/pix2pix/infer.py -------------------------------------------------------------------------------- /Vision/gan/pix2pix/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/pix2pix/infer.sh -------------------------------------------------------------------------------- /Vision/gan/pix2pix/models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/pix2pix/models/networks.py -------------------------------------------------------------------------------- /Vision/gan/pix2pix/requirements.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | numpy 3 | Pillow -------------------------------------------------------------------------------- /Vision/gan/pix2pix/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/pix2pix/train.py -------------------------------------------------------------------------------- /Vision/gan/pix2pix/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/pix2pix/train.sh -------------------------------------------------------------------------------- /Vision/gan/pix2pix/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/pix2pix/utils/data_utils.py -------------------------------------------------------------------------------- /Vision/gan/pix2pix/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/pix2pix/utils/plot.py -------------------------------------------------------------------------------- /Vision/gan/pix2pix/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/gan/pix2pix/utils/utils.py -------------------------------------------------------------------------------- /Vision/oneflow_face/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/README.md -------------------------------------------------------------------------------- /Vision/oneflow_face/README_CH.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/README_CH.md -------------------------------------------------------------------------------- /Vision/oneflow_face/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/backbones/__init__.py -------------------------------------------------------------------------------- /Vision/oneflow_face/backbones/ir_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/backbones/ir_resnet.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/base.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/glint360k_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/glint360k_mbf.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/glint360k_r100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/glint360k_r100.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/glint360k_r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/glint360k_r18.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/glint360k_r34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/glint360k_r34.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/glint360k_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/glint360k_r50.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/ms1mv3_mbf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/ms1mv3_mbf.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/ms1mv3_r18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/ms1mv3_r18.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/ms1mv3_r34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/ms1mv3_r34.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/ms1mv3_r50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/ms1mv3_r50.py -------------------------------------------------------------------------------- /Vision/oneflow_face/configs/speed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/configs/speed.py -------------------------------------------------------------------------------- /Vision/oneflow_face/convert.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/convert.sh -------------------------------------------------------------------------------- /Vision/oneflow_face/eval/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/oneflow_face/eval/onnx_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/eval/onnx_helper.py -------------------------------------------------------------------------------- /Vision/oneflow_face/eval/onnx_ijbc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/eval/onnx_ijbc.py -------------------------------------------------------------------------------- /Vision/oneflow_face/eval/verification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/eval/verification.py -------------------------------------------------------------------------------- /Vision/oneflow_face/function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/function.py -------------------------------------------------------------------------------- /Vision/oneflow_face/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/graph.py -------------------------------------------------------------------------------- /Vision/oneflow_face/oneflow2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/oneflow2onnx.py -------------------------------------------------------------------------------- /Vision/oneflow_face/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/requirements.txt -------------------------------------------------------------------------------- /Vision/oneflow_face/tools/mx_recordio_2_ofrecord.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/tools/mx_recordio_2_ofrecord.py -------------------------------------------------------------------------------- /Vision/oneflow_face/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/train.py -------------------------------------------------------------------------------- /Vision/oneflow_face/train_ddp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/train_ddp.sh -------------------------------------------------------------------------------- /Vision/oneflow_face/train_graph_distributed.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/train_graph_distributed.sh -------------------------------------------------------------------------------- /Vision/oneflow_face/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/oneflow_face/utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/utils/losses.py -------------------------------------------------------------------------------- /Vision/oneflow_face/utils/ofrecord_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/utils/ofrecord_data_utils.py -------------------------------------------------------------------------------- /Vision/oneflow_face/utils/utils_callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/utils/utils_callbacks.py -------------------------------------------------------------------------------- /Vision/oneflow_face/utils/utils_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/utils/utils_config.py -------------------------------------------------------------------------------- /Vision/oneflow_face/utils/utils_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/utils/utils_logging.py -------------------------------------------------------------------------------- /Vision/oneflow_face/val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/val.py -------------------------------------------------------------------------------- /Vision/oneflow_face/val.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/oneflow_face/val.sh -------------------------------------------------------------------------------- /Vision/reid/BoT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/README.md -------------------------------------------------------------------------------- /Vision/reid/BoT/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/data_loader.py -------------------------------------------------------------------------------- /Vision/reid/BoT/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/infer.sh -------------------------------------------------------------------------------- /Vision/reid/BoT/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/loss.py -------------------------------------------------------------------------------- /Vision/reid/BoT/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/lr_scheduler.py -------------------------------------------------------------------------------- /Vision/reid/BoT/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/model.py -------------------------------------------------------------------------------- /Vision/reid/BoT/reid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/reid.py -------------------------------------------------------------------------------- /Vision/reid/BoT/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/requirements.txt -------------------------------------------------------------------------------- /Vision/reid/BoT/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/train.sh -------------------------------------------------------------------------------- /Vision/reid/BoT/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/reid/BoT/utils/distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/utils/distance.py -------------------------------------------------------------------------------- /Vision/reid/BoT/utils/loggers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/reid/BoT/utils/loggers.py -------------------------------------------------------------------------------- /Vision/segmentation/FODDet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FODDet/README.md -------------------------------------------------------------------------------- /Vision/segmentation/FODDet/model/UNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FODDet/model/UNet.py -------------------------------------------------------------------------------- /Vision/segmentation/FODDet/requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | opencv-python 3 | matplotlib 4 | albumentations 5 | tqdm -------------------------------------------------------------------------------- /Vision/segmentation/FODDet/test/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FODDet/test/dataloader.py -------------------------------------------------------------------------------- /Vision/segmentation/FODDet/test/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FODDet/test/test.py -------------------------------------------------------------------------------- /Vision/segmentation/FODDet/test/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FODDet/test/test.sh -------------------------------------------------------------------------------- /Vision/segmentation/FODDet/test/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FODDet/test/visualize.py -------------------------------------------------------------------------------- /Vision/segmentation/FODDet/train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FODDet/train/train.py -------------------------------------------------------------------------------- /Vision/segmentation/FODDet/train/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FODDet/train/train.sh -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FaceSeg/README.md -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FaceSeg/infer.py -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FaceSeg/infer.sh -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/models/LinkNet34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FaceSeg/models/LinkNet34.py -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/models/resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FaceSeg/models/resnet50.py -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FaceSeg/requirements.txt -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FaceSeg/train.py -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FaceSeg/train.sh -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/utils/numpy_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FaceSeg/utils/numpy_data_utils.py -------------------------------------------------------------------------------- /Vision/segmentation/FaceSeg/validation_oneflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/FaceSeg/validation_oneflow.py -------------------------------------------------------------------------------- /Vision/segmentation/U-Net/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/U-Net/README.md -------------------------------------------------------------------------------- /Vision/segmentation/U-Net/TrainUnetDataSet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/U-Net/TrainUnetDataSet.py -------------------------------------------------------------------------------- /Vision/segmentation/U-Net/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/U-Net/dataloader.py -------------------------------------------------------------------------------- /Vision/segmentation/U-Net/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/U-Net/log.py -------------------------------------------------------------------------------- /Vision/segmentation/U-Net/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/U-Net/plot.py -------------------------------------------------------------------------------- /Vision/segmentation/U-Net/predict_unet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/U-Net/predict_unet_test.py -------------------------------------------------------------------------------- /Vision/segmentation/U-Net/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/U-Net/requirements.txt -------------------------------------------------------------------------------- /Vision/segmentation/U-Net/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/U-Net/test.sh -------------------------------------------------------------------------------- /Vision/segmentation/U-Net/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/U-Net/train.sh -------------------------------------------------------------------------------- /Vision/segmentation/U-Net/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/segmentation/U-Net/unet.py -------------------------------------------------------------------------------- /Vision/style_transform/fast_neural_style/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/style_transform/fast_neural_style/README.md -------------------------------------------------------------------------------- /Vision/style_transform/fast_neural_style/infer.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/style_transform/fast_neural_style/infer.sh -------------------------------------------------------------------------------- /Vision/style_transform/fast_neural_style/requirements.txt: -------------------------------------------------------------------------------- 1 | opencv_python 2 | numpy 3 | -------------------------------------------------------------------------------- /Vision/style_transform/fast_neural_style/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/Vision/style_transform/fast_neural_style/train.sh -------------------------------------------------------------------------------- /ci/check/run_py_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ci/check/run_py_format.py -------------------------------------------------------------------------------- /dev-requirements.txt: -------------------------------------------------------------------------------- 1 | black==19.10b0 ; python_version >= "3.6" 2 | -------------------------------------------------------------------------------- /ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/__init__.py -------------------------------------------------------------------------------- /ops/csrc/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/csrc/CMakeLists.txt -------------------------------------------------------------------------------- /ops/csrc/nms.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/csrc/nms.cpp -------------------------------------------------------------------------------- /ops/csrc/nms.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/csrc/nms.cu -------------------------------------------------------------------------------- /ops/csrc/roi_align.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/csrc/roi_align.cpp -------------------------------------------------------------------------------- /ops/csrc/roi_align.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/csrc/roi_align.cu -------------------------------------------------------------------------------- /ops/csrc/roi_align.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/csrc/roi_align.h -------------------------------------------------------------------------------- /ops/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/nms.py -------------------------------------------------------------------------------- /ops/roi_align.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/roi_align.py -------------------------------------------------------------------------------- /ops/test/nms_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/test/nms_test.py -------------------------------------------------------------------------------- /ops/test/roi_align_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/ops/test/roi_align_test.py -------------------------------------------------------------------------------- /scripts/compare_speed_with_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/scripts/compare_speed_with_pytorch.py -------------------------------------------------------------------------------- /scripts/swin_dataloader_compare_speed_with_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Oneflow-Inc/models/HEAD/scripts/swin_dataloader_compare_speed_with_pytorch.py --------------------------------------------------------------------------------