Where is the loss calculated?

[Alireza Khorshidi]

Here is where the loss is calculated for the case of single core:

https://bitbucket.org/andrewpeterson/amp/src/11ad04a3b6c658e183b9cc1b90fc71c7c298115c/model/init.py?at=master&fileviewer=file-view-default#init.py-199

(Line 199 of model/init.py)

Though I am not seeing where the loss is being calculated in the case of multiple cores here?

https://bitbucket.org/andrewpeterson/amp/src/11ad04a3b6c658e183b9cc1b90fc71c7c298115c/model/init.py?at=master&fileviewer=file-view-default#init.py-211

(Line 211 of model/init.py)

[andrew_peterson]

It's the same line for both cases! That is, see line 78 of main.py:

[Alireza Khorshidi]

I understood it.