<DLRMv2> Add return of construct_model in dlrm_main.py

.dockerignore

@@ -0,0 +1,2 @@
+.git
+.github

.gitignore

@@ -3,6 +3,7 @@
 *.pyc
 .DS_Store
 **.log
+pretrained/
 .pytest*
 .venv*
 .coverage
@@ -16,4 +17,5 @@ output/
 tools/docker/models*
 .ipynb_checkpoints
 nc_workspace
-
+benchmarks/horovod
+cloud_data_connector/credentials.json

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "models/aidd/pytorch/alphafold2/inference/alphafold"]
+	path = models/aidd/pytorch/alphafold2/inference/alphafold
+	url = https://github.com/deepmind/alphafold

CODEOWNERS

@@ -3,12 +3,14 @@
 
 # These owners will be the default owners for everything in the repo,
 # but PR owner should be able to assign other contributors when appropriate
-*    @ashahba @claynerobison @dmsuehir
+* [email protected] @ashahba @claynerobison 
 datasets @ashahba @claynerobison @dzungductran
 docs @claynerobison @mhbuehler
-k8s  @ashahba @dzungductran @kkasravi
-models @agramesh1 @ashraf-bhuiyan @riverliuintel @wei-v-wang
+k8s  @ashahba @dzungductran
+models @ashraf-bhuiyan @riverliuintel
+models @riverliuintel
 models/**/pytorch/ @leslie-fang-intel @jiayisunx @zhuhaozhe
+quickstart [email protected]
 quickstart/**/pytorch/ @leslie-fang-intel @jiayisunx @zhuhaozhe
 
 # Order is important. The last matching pattern has the most precedence.

Makefile

@@ -30,7 +30,7 @@ all: venv lint unit_test
 $(ACTIVATE):
 	@echo "Updating virtualenv dependencies in: $(VIRTUALENV_DIR)..."
 	@test -d $(VIRTUALENV_DIR) || $(VIRTUALENV_EXE) $(VIRTUALENV_DIR)
-	@. $(ACTIVATE) && python -m pip install -r requirements-test.txt
+	@. $(ACTIVATE) && python -m pip install -r requirements.txt
 	@touch $(ACTIVATE)
 
 venv: $(ACTIVATE)
@@ -53,6 +53,20 @@ unit_test:
 	@echo "Running unit tests..."
 	tox -e py3-py.test
 
+test_tl_tf_notebook: venv
+	@. $(ACTIVATE) && pip install -r docs/notebooks/transfer_learning/requirements.txt && \
+	bash docs/notebooks/transfer_learning/run_tl_notebooks.sh $(CURDIR)/docs/notebooks/transfer_learning/text_classification/tfhub_bert_text_classification/BERT_Binary_Text_Classification.ipynb remove_for_custom_dataset && \
+	bash docs/notebooks/transfer_learning/run_tl_notebooks.sh $(CURDIR)/docs/notebooks/transfer_learning/image_classification/tf_image_classification/Image_Classification_Transfer_Learning.ipynb remove_for_custom_dataset && \
+	bash docs/notebooks/transfer_learning/run_tl_notebooks.sh $(CURDIR)/docs/notebooks/transfer_learning/image_classification/huggingface_image_classification/HuggingFace_Image_Classification_Transfer_Learning.ipynb && \
+	bash docs/notebooks/transfer_learning/run_tl_notebooks.sh $(CURDIR)/docs/notebooks/transfer_learning/question_answering/BERT_Question_Answering.ipynb && \
+	bash docs/notebooks/transfer_learning/run_tl_notebooks.sh $(CURDIR)/docs/notebooks/transfer_learning/text_classification/tfhub_bert_text_classification/BERT_Multi_Text_Classification.ipynb remove_for_custom_dataset
+
+test_tl_pyt_notebook: venv
+	@. $(ACTIVATE) && pip install -r docs/notebooks/transfer_learning/requirements.txt && \
+	bash docs/notebooks/transfer_learning/run_tl_notebooks.sh $(CURDIR)/docs/notebooks/transfer_learning/image_classification/pytorch_image_classification/PyTorch_Image_Classification_Transfer_Learning.ipynb remove_for_custom_dataset && \
+	bash docs/notebooks/transfer_learning/run_tl_notebooks.sh $(CURDIR)/docs/notebooks/transfer_learning/object_detection/pytorch_object_detection/PyTorch_Object_Detection_Transfer_Learning.ipynb remove_for_custom_dataset && \
+	bash docs/notebooks/transfer_learning/run_tl_notebooks.sh $(CURDIR)/docs/notebooks/transfer_learning/text_classification/pytorch_text_classification/PyTorch_Text_Classifier_fine_tuning.ipynb remove_for_custom_dataset
+
 test: lint unit_test
 
 clean: