search

ZhiGroup / Med-BERT

Med-BERT, contextualized embedding model for structured EHR data
Apache License 2.0
260 stars 63 forks source link

Pre-training error #10

Closed koustav123 closed 1 year ago

koustav123 commented 2 years ago

Hi I created a toy data with 1000 patients, following the sample data provided. My cohort looks like this. image

The scripts for partition and TF extraction went fine. But when I start the PT script, it throws the following error.

image

I will really appreciate if you have any suggestions, off the top of your head.

lrasmy commented 2 years ago

Hi Koustav Ghosal,

This error is probably due to the TensorFlow version and compatibility with your Cuda. This code version is mainly compatible with Tensorflow 1.15 or less.

you can follow the solution on https://developpaper.com/install-tensorflow-1-15-in-python-3-8-cuda-11-environment/

Thanks, Laila