Fix ARGD

[caio-freitas]

• [x] Fix training, validation isn't matching
• [x] Fix sampling, despite small losses actions are weird
• [x] Make sure node ID Embedding is correct on ARGD