(model_output, gt)
| 3 | |
| 4 | |
| 5 | def function_mse(model_output, gt): |
| 6 | idx = model_output['model_in']['idx'].long().squeeze() |
| 7 | loss = (model_output['model_out']['output'][:, idx] - gt['func'][:, idx]) ** 2 |
| 8 | return {'func_loss': loss.mean()} |
| 9 | |
| 10 | |
| 11 | def image_mse(model_output, gt): |
nothing calls this directly
no outgoing calls
no test coverage detected