lccasagrande / Deep-Knowledge-Tracing

An implementation of the Deep Knowledge Tracing (DKT) using Tensorflow 2.0
MIT License
94 stars 38 forks source link

with Gpu fit #10

Closed RGalilea closed 3 years ago

RGalilea commented 3 years ago

with tf.device(tf.test.gpu_device_name()):