Estimate total GPU memory required for a neural network given parameter count, batch size, activation memory, and optimizer state.
Enter the values for the patient or scenario you are assessing. Estimate total GPU memory required for a neural network given parameter count, batch size, activation memory, and optimizer state. Use the nn memory result to inform your clinical assessment.