Pytorch based reimplementation of COMS: Conservative Objective Models for Effective Offline Model-Based Optimization.

The original tensorflow implementation is here.

run_coms.py is the entry point of the code.

pytorch_nets.py contains the surrogate model.

pytorch_coms.pycontains the training algorithm.