search

MIC-DKFZ / BraTS2017

Apache License 2.0
74 stars 16 forks source link

Can I accelerate the training course with GPU? #21

Closed yianzhongguo closed 5 years ago

yianzhongguo commented 5 years ago

@FabianIsensee Hi, sir. I am training the network of your code, but it is too slow as follows. Can I accelerate it with GPU? Thank you!

val dice: [0.81688768 0.68206574 0.63964228 0.01935757] This epoch took 19384.164900 sec

FabianIsensee commented 5 years ago

Of course! http://deeplearning.net/software/theano_versions/dev/library/config.html

You need to set the device in theanorc