# DRiLLS

**Repository Path**: qifeng-dai/DRiLLS

## Basic Information

- **Project Name**: DRiLLS
- **Description**: 强化学习求解逻辑优化问题。
- **Primary Language**: Unknown
- **License**: BSD-3-Clause
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-02-21
- **Last Updated**: 2025-03-12

## Categories & Tags

**Categories**: Uncategorized

**Tags**: AI

## README

## 介绍
本项目克隆自[DRiLLS](https://github.com/scale-lab/DRiLLS)。

相比而言有几大改进：
1. 增加更多算子
2. 优化`state`的设计，让RL算法更加稳定
3. 采用`torch`替换`tensorflow`，并优化神经网络结构
4. 优化性能，提升了`DRiLLS`的训练速度和精度
5. 支持`GPU`
6. 修复了一些bug

## Paper
参见：[DRiLLS Paper](https://arxiv.org/abs/1911.04021)。

## 安装
需要安装有：`Python 3.10`, `pip3`, `pyenv`。

关于 `pyenv` 的安装，请参考：[pyenv](https://github.com/pyenv/pyenv)。

安装依赖：`pip install -r requirements.txt`

## 运行

1. 编辑 `params.yml` 文件
2. 训练 `python drills.py -m train [-p params.yml] [-l]`
3. 优化 `python drills.py -m optimize [-p params.yml] [-l]`

## 架构
<img src="doc/img/drills-architecture.png" width="70%" style="display: block;  margin: 0 auto;">