-
RLlib ModelV2 API is ready to use. It improves support for Keras and RNN models, as well as allowing object-oriented reuse of variables. ModelV1 API is deprecated. No migration is needed.
-
ray.experimental.sgd.pytorch.PyTorchTraineris ready for early adopters. Checkout the doc here and we welcome your feedback!
model_creator = lambda config: YourPyTorchModel()
data_creator = lambda config: YourTrainingSet(), YourValidationSet()