mind2cloud commited on
Commit
c0d2305
·
verified ·
1 Parent(s): 3023ad8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -4,6 +4,8 @@ base_model: deepseek-ai/DeepSeek-R1-Distill-Qwen-7B
4
  library_name: peft
5
  ---
6
 
 
 
7
  The DeepSeek-R1-Distill-Qwen-7B model has been fine-tuned **to predict hyperparameters for neural network models**. Leveraging the power of large language models (LLMs), this version can analyze neural network architectures and generate optimal hyperparameter configurations — such as learning rate, batch size, dropout, momentum, and so on — for a given task. This approach offers a competitive alternative to traditional optimization methods like the Optuna Framework.
8
 
9
 
@@ -16,8 +18,7 @@ The DeepSeek-R1-Distill-Qwen-7B model has been fine-tuned **to predict hyperpara
16
 
17
  A large language model used in the <a href='https://github.com/ABrain-One/NN-GPT'>NNGPT</a> project for generating training hyperparameters for neural networks from the <a href='https://github.com/ABrain-One/NN-Dataset'>LEMUR NN Dataset</a>
18
 
19
- # Model Card for DeepSeek-R1-Distill-Qwen-7B-R
20
-
21
  This repository provides a **fine-tuned version** of [deepseek-ai/DeepSeek-R1-Distill-Qwen-7B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-7B) using the [PEFT](https://github.com/huggingface/peft) library with LoRA. The final model is **merged** so it can be loaded in one step via:
22
  ```python
23
  from transformers import AutoModelForCausalLM, AutoTokenizer
 
4
  library_name: peft
5
  ---
6
 
7
+ # DeepSeek-R1-Distill-Qwen-7B-R
8
+
9
  The DeepSeek-R1-Distill-Qwen-7B model has been fine-tuned **to predict hyperparameters for neural network models**. Leveraging the power of large language models (LLMs), this version can analyze neural network architectures and generate optimal hyperparameter configurations — such as learning rate, batch size, dropout, momentum, and so on — for a given task. This approach offers a competitive alternative to traditional optimization methods like the Optuna Framework.
10
 
11
 
 
18
 
19
  A large language model used in the <a href='https://github.com/ABrain-One/NN-GPT'>NNGPT</a> project for generating training hyperparameters for neural networks from the <a href='https://github.com/ABrain-One/NN-Dataset'>LEMUR NN Dataset</a>
20
 
21
+ # How to Use
 
22
  This repository provides a **fine-tuned version** of [deepseek-ai/DeepSeek-R1-Distill-Qwen-7B](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-7B) using the [PEFT](https://github.com/huggingface/peft) library with LoRA. The final model is **merged** so it can be loaded in one step via:
23
  ```python
24
  from transformers import AutoModelForCausalLM, AutoTokenizer