Transformer-BiGRU

The codebase and result for using BiGRU to predict GPU memory usage in deep learning training.

Name		Name	Last commit message	Last commit date
.gitignore		.gitignore
Data-1.xlsx		Data-1.xlsx
LICENSE		LICENSE
README.md		README.md
README_pytorch.md		README_pytorch.md
confusion_matrix_confusion_matrix_for_test_data.png		confusion_matrix_confusion_matrix_for_test_data.png
confusion_matrix_confusion_matrix_for_training_data.png		confusion_matrix_confusion_matrix_for_training_data.png
flip_layer.py		flip_layer.py
main.py		main.py
model.py		model.py
model_1.py		model_1.py
model_visualization.png		model_visualization.png
pam_visualization.png		pam_visualization.png
polygon_area_metric.py		polygon_area_metric.py
prediction_results_test_set_prediction_results.png		prediction_results_test_set_prediction_results.png
prediction_results_training_set_prediction_results.png		prediction_results_training_set_prediction_results.png
pytorch_main.py		pytorch_main.py
requirements.txt		requirements.txt
roc_curve.png		roc_curve.png
setup_env.sh		setup_env.sh
training_progress.png		training_progress.png

Navigation Menu