Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .devcontainer/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
# Image for a Python 3 development environment
FROM python:3.8-slim
FROM registry.access.redhat.com/ubi8/python-39:1-105

# Add any tools that are needed beyond Python
RUN apt-get update && \
Expand Down
2 changes: 1 addition & 1 deletion entity_standardizer/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ Entity standardizer package contains the following two single-shot models:
2. *siamese* - A supervised approach that uses BERT models connected in a Siamese network. The default model is the siamese model.

Use the following steps to install, train, and inference with the entity standardizer models as a standalone package.
The entity standardizer package requires python >= 3.6 environment.
The entity standardizer package requires python >= 3.6 environment. To train the model, switch to the gpu version of pytorch. Currently the cpu version is used and it can be modified in the requirements.txt under the entity_standardizer folder.

1. Run *setup.sh* to install dependencies and entity standardizer package
```
Expand Down