5,337
edits
Debugging ML Models: Difference between revisions
no edit summary
No edit summary |
|||
| Line 5: | Line 5: | ||
** If it doesn't overfit, there may be an issue with your code. | ** If it doesn't overfit, there may be an issue with your code. | ||
** You can try increasing the capacity (e.g number of filters or number of nodes in FC) 2-4x. | ** You can try increasing the capacity (e.g number of filters or number of nodes in FC) 2-4x. | ||
**: | **: If the input is 3 channels, then the first conv layer should have more than 3 channels. | ||
** Check that your loss is implemented correctly and taken against the correct ground truth image. | ** Check that your loss is implemented correctly and taken against the correct ground truth image. | ||
* Dump all inputs and outputs into [[TensorBoard]]. You may have an unexpected input or output somewhere. | * Dump all inputs and outputs into [[TensorBoard]]. You may have an unexpected input or output somewhere. | ||