omega/model/mlp_weights.pth