search

Pongpisit-Thanasutives / CSRNet-Keras-train

5 stars 1 forks source link

Kernel problems #1

Open stardust3dd opened 5 years ago

stardust3dd commented 5 years ago

While running the fitting function, the kernel keeps stopping every time in the first epoch itself without any explanation or error messages. Any idea why? @Pongpisit-Thanasutives please respond.

Pongpisit-Thanasutives commented 5 years ago

Did you use a GPU for training the model?

Pongpisit-Thanasutives commented 5 years ago

For more information in order to run the code successfully, I encourage you to visit https://github.com/Neerajj9/CSRNet-keras. the train_model.py is just for the training process.

stardust3dd commented 5 years ago

Thanks for responding. Could you lease provide the specifications of your GPU and your system? Did you use CUDA? I'm trying to check if the problem lies within my system configuration.