https://github.com/adalca/neurite/blob/46c975e589e063a8ff88b5c115d19f73a956ebf8/neurite/tf/models.py#L1043
neurite/neurite/tf/models.py
Line 1043 in 46c975e