Wancong (Kevin) Zhang

Hi there, Thanks for sharing your work. On the bottom of your README, you said you have already "Modify wraprnnt-pytorch to compatible with apex mixed precision". But I noticed that...

Hi, quick question, I noticed that you saved the pytorch weights for joint network and prediction network, and then reloaded them manually using cpp. Why did you choose this way...

How to get replays for training UL?

2

Hi, In the [rlpyt/rlpyt/ul](https://github.com/astooke/rlpyt/tree/master/rlpyt/ul) directory, description says: "See "experiments" folder for scripts to run online RL agents with ATC ... and **other RL agents to gather expert demonstrations**. However, I...

How to call Model.score inside my custom CPP binding module

1

Hi, I have 2 questions: 1) I am interested in calling model.Score() inside my own PYBIND11_MODULE, like this: ``` #include "/home/cirrascale/kevin/kenlm/lm/model.hh" #include using namespace std; using namespace torch::indexing; int test_function(torch::Tensor...

mismatch between paper and repo hyperparameters

1

Hi, There seems to be at least two mismatch between the paper and repo hyperparameters. 1) SPR_weight In paper: "We set λ SPR = 2 and λ IM = 1...