search

zilongzhong / SSRN

This is a tensorflow and keras based implementation of SSRNs in the IEEE T-GRS paper "Spectral-Spatial Residual Network for Hyperspectral Image Classification: A 3-D Deep Learning Framework".
Other
211 stars 74 forks source link

I can only run one epoch by running `SSRN_IN.py` #18

Closed darrenzhang1007 closed 4 years ago

darrenzhang1007 commented 4 years ago

Hello teacher, I meet a problem. I use Python 3.6 + keras-2.2.4 . I can only run one epoch when run SSRN_IN.py, Then the code was interrupted. But there is no error, 微信截图_20190926151727 how can i to revised it? Thanks. @zilongzhong

zilongzhong commented 4 years ago

I can run the codes without such a problem. Make sure you have the same conda environment setting.

Best regards, Zilong

On Sep 26, 2019, at 3:15 PM, DarrenmondZhang notifications@github.com wrote:

I can only run one epoch when run SSRN_IN.py, Then the code was interrupted. But there is no error,

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or mute the thread.

darrenzhang1007 commented 4 years ago

Before, I used the environment: cudn10+tensorflow14.It didn`t work. Now, I set my environment: cudn9.2+tensorflow1.8. This problem can solve. Thanks,