Skip to content
Snippets Groups Projects
Commit c46d0810 authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu :desktop:
Browse files

brought age delta calculation as output-target, in line with papers

parent b6098c7f
No related branches found
No related merge requests found
......@@ -213,7 +213,7 @@ def _statistics_calculator(output_age, target_age):
output_age = np.array(output_age)
target_age = np.array(target_age)
age_delta = target_age - output_age
age_delta = output_age - target_age
loss = MSELoss(torch.from_numpy(output_age), torch.from_numpy(target_age)).numpy()
return age_delta, loss
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment