Deep learning for brain metastasis detection and segmentation in longitudinal MRI data
BSD 3-Clause "New" or "Revised" License
2
stars
1
forks
source link
set memory growth flag to True to prevent memory errors #10
Closed
damiankucharski closed 1 year ago
This piece of code should prevent some memory errors as I have experienced when running training. Note that it works only with tensorflow 2.2+