Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

Export memories to the observation output #4040

Merged
merged 3 commits into from
Oct 19, 2021
Merged

Export memories to the observation output #4040

merged 3 commits into from
Oct 19, 2021

Conversation

apravesh
Copy link
Contributor

Export memories to the observation output. Want to use it to create demos in fbcode.

Sample output:

    "id": "BlenderBot2Fid",
    "episode_done": false,
    "text": " That sounds like a good job. How long have you been working there? Do you like it?",
    "beam_texts": [
      [
        " That sounds like a good job. How long have you been working there? Do you like it?",
        -7.5464444160461
      ],
      [
        " That sounds like a good job. How long have you been working there? Do you enjoy it?",
        -7.8649473190308
      ],
      [
        " That sounds like a good job. How long have you been working there? Do you like your job?",
        -8.3311004638672
      ],
      [
        " That sounds like a good job. How long have you been working there? Do you enjoy your job?",
        -8.4195833206177
      ],
      [
        " That sounds like a good job. How long have you been working there? Do you have any pets?",
        -8.5225353240967
      ],
      [
        " That sounds like a good job. How long have you been working there? Do you like your coworkers?",
        -8.5445289611816
      ],
      [
        " That sounds like a good job. How long have you been working there? Do you have a family?",
        -8.7001924514771
      ],
      [
        " That sounds like a good job. How long have you been working there? Do you have any pets at home?",
        -9.0912303924561
      ],
      [
        " That sounds like a good job. How long have you been working there? Do you have any plans to travel?",
        -9.7705593109131
      ],
      [
        " That sounds like a good job. How long have you been working there? Do you have any plans to leave?",
        -9.817099571228
      ]
    ],
    "memories": [
      "partner's persona: I am a software engineer. I work at Facebook.",
      "your persona: I work at Walmart. I am a cashier.",
      "partner's persona: My name is Pravesh. I am a male."
    ],
    "thread_id": 2345
  }```

Export memories to the observation output. Want to use it to create demos in fbcode.
Copy link
Contributor

@klshuster klshuster left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this looks good! you'll just need to update the code to use self.model_api instead of self.model (this is what your merge conflict is)

Copy link
Contributor

@klshuster klshuster left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

looks great!

@apravesh apravesh merged commit d5a7633 into main Oct 19, 2021
@apravesh apravesh deleted the memories_export branch October 19, 2021 17:13
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants