diff --git a/demo_api/dual_bert/model_card/dual_bert.json b/demo_api/dual_bert/model_card/dual_bert.json index 6f8819e..e2ff465 100644 --- a/demo_api/dual_bert/model_card/dual_bert.json +++ b/demo_api/dual_bert/model_card/dual_bert.json @@ -27,18 +27,15 @@ "Test results for Stance classification (SemEval-2017 Dataset):", "Single Stance Type Evaluation (Trained/Reported): Support-F1: 0.421/ Deny-F1: 0.255/ Query-F1: 0.52/ Comment-F1: 0.841", "Overall Evaluation (Trained/Reported): Macro-F1: 0.509/ Accuracy: 0.763/" - ], - "trainingConfig": { - "text": "Not available." - }, + "trainingConfig": "Not available.", "trainingTime": "~2 hours for 100 epochs on a single V100 GPU.", "modelWeights": { "text": "https://storage.googleapis.com/sgnlp/models/dual_bert/pytorch_model.bin", "url": "https://storage.googleapis.com/sgnlp/models/dual_bert/pytorch_model.bin" }, "modelConfig": { - "text": "", + "text": "NA", "url": "" }, "modelInput": "Conversation threads", diff --git a/polyaxon/coupled_hierarchical_transformer/conda.yml b/polyaxon/dual_bert/conda.yml similarity index 100% rename from polyaxon/coupled_hierarchical_transformer/conda.yml rename to polyaxon/dual_bert/conda.yml diff --git a/polyaxon/coupled_hierarchical_transformer/experiment.Dockerfile b/polyaxon/dual_bert/experiment.Dockerfile similarity index 87% rename from polyaxon/coupled_hierarchical_transformer/experiment.Dockerfile rename to polyaxon/dual_bert/experiment.Dockerfile index a5e4906..c67890b 100644 --- a/polyaxon/coupled_hierarchical_transformer/experiment.Dockerfile +++ b/polyaxon/dual_bert/experiment.Dockerfile @@ -9,7 +9,7 @@ RUN rm /bin/sh && ln -s /bin/bash /bin/sh && \ WORKDIR $WORK_DIR USER $USER -COPY build/polyaxon/coupled_hierarchical_transformer/conda.yml . +COPY build/polyaxon/dual_bert/conda.yml . RUN conda env update -f conda.yml -n base WORKDIR /code diff --git a/polyaxon/coupled_hierarchical_transformer/train_multitask.yml b/polyaxon/dual_bert/train_multitask.yml similarity index 100% rename from polyaxon/coupled_hierarchical_transformer/train_multitask.yml rename to polyaxon/dual_bert/train_multitask.yml