A dockerized flask API, serving ALBERT and BERT predictions using TensorFlow 2.0.
Stars
32
Forks
5
Watchers
32
Open Issues
1
Overall repository health assessment
No package.json found
This might not be a Node.js project
11
commits
1
commits
Fixed a bug with loading albert models in API image. Minor refactors and cleanup.
10c48b6View on GitHubAdded support for using ALBERT models. Improved and overhauled data loading, tokenization and pretrained model retrieval.
b6bc898View on GitHubMerge pull request #4 from sarnikowski/overhaul/tf2
aa5e68bView on GitHubMerge pull request #2 from sarnikowski/feature/makefile
edcda9cView on GitHubUpdated Dockerfile to use GPU. Makefile base model target has been parametrized. Training and prediction scripts have been upgraded to TF 2.0. Updated methods in utils and trimmed some fat. Updated README
a4473a8View on GitHubWork in progress on training script. Implemented main blocks, however is untested. Logging needs refactoring. Removed some unecessary utils and requirements.
2e6f6adView on GitHubInitial README draft for TF2 implementation. Still work in progress
30319c5View on GitHubCode is now runnable using the make targets. Added stack overflow data
90c293bView on GitHubMerge pull request #1 from Ayush-iitkgp/Add-seaborn-in-requirements.txt
f20093bView on GitHub