This model uses GPT to model (Return_1, state_1, action_1, Return_2, state_2, ...)

empirical moving average

This model uses GPT to model (Return_1, state_1, action_1, Return_2, state_2, ...)

Base class for outputs of models predicting if two sentences are consecutive or not. Args: loss (:obj:`torch.FloatTensor` of shape :o

An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.