musically-ut / tf_rmtpp

Recurrent Marked Temporal Point Processes
MIT License
55 stars 16 forks source link

Allow running on CPUs only. #2

Closed musically-ut closed 6 years ago

musically-ut commented 6 years ago

Will allow for easier debugging and testing.

However, the behavior of embedding_lookup() on CPUs is less forgiving than on GPUs (where it merely returns 0 for out of bound accesses).