{"id":19421,"date":"2025-01-26T22:13:30","date_gmt":"2025-01-26T14:13:30","guid":{"rendered":"https:\/\/www.aisharenet.com\/?p=19421"},"modified":"2025-01-26T22:13:30","modified_gmt":"2025-01-26T14:13:30","slug":"tinyzero","status":"publish","type":"post","link":"https:\/\/www.kdjingpai.com\/en\/tinyzero\/","title":{"rendered":"TinyZero\uff1a\u4f4e\u6210\u672c\u590d\u73b0 DeepSeeK-R1 Zero \u7684\u987f\u609f\u6548\u679c"},"content":{"rendered":"<p>TinyZero \u662f\u4e00\u4e2a\u57fa\u4e8e veRL \u7684\u5f3a\u5316\u5b66\u4e60\u6a21\u578b\uff0c\u65e8\u5728\u590d\u73b0 <a href=\"https:\/\/www.kdjingpai.com\/en\/deepseek-r1nenglixiang\/\">DeepSeeK-R1<\/a> Zero \u5728\u5012\u8ba1\u65f6\u548c\u4e58\u6cd5\u4efb\u52a1\u4e2d\u7684\u8868\u73b0\u3002\u4ee4\u4eba\u60ca\u8bb6\u7684\u662f\uff0c\u8be5\u9879\u76ee\u4ec5\u9700 30 \u7f8e\u5143\u7684\u8fd0\u884c\u6210\u672c\uff08\u4f7f\u7528 2xH200\uff0c\u6bcf\u5c0f\u65f6 6.4 \u7f8e\u5143\uff0c\u4e0d\u5230 5 \u5c0f\u65f6\uff09\uff0c\u5c31\u80fd\u5b9e\u73b0\u4e0e DeepSeeK-R1 Zero \u76f8\u540c\u7684\u987f\u609f\u6548\u679c\u3002\u901a\u8fc7\u5f3a\u5316\u5b66\u4e60\uff08RL\uff09\uff0c3B \u57fa\u7840\u8bed\u8a00\u6a21\u578b\uff08LM\uff09\u80fd\u591f\u81ea\u4e3b\u5f00\u53d1\u81ea\u6211\u9a8c\u8bc1\u548c\u641c\u7d22\u80fd\u529b\u3002\u7528\u6237\u53ef\u4ee5\u901a\u8fc7\u7b80\u5355\u7684\u5b89\u88c5\u6b65\u9aa4\u548c\u8bad\u7ec3\u8fc7\u7a0b\uff0c\u4f53\u9a8c\u5230 TinyZero \u7684\u5f3a\u5927\u529f\u80fd\u548c\u521b\u65b0\u6027\u3002<\/p>\n<p><img loading=\"lazy\" decoding=\"async\" class=\"\" title=\"TinyZero\uff1a\u4f4e\u6210\u672c\u590d\u73b0 DeepSeeK-R1 Zero \u7684\u987f\u609f\u6548\u679c-1\" src=\"https:\/\/www.kdjingpai.com\/wp-content\/uploads\/2025\/01\/1330b89e947f13b.jpg\" alt=\"TinyZero\uff1a\u4f4e\u6210\u672c\u590d\u73b0 DeepSeeK-R1 Zero \u7684\u987f\u609f\u6548\u679c-1\" width=\"713\" height=\"426\" \/><\/p>\n<p>&nbsp;<\/p>\n<h2>\u529f\u80fd\u5217\u8868<\/h2>\n<ul>\n<li><strong>\u5012\u8ba1\u65f6\u4efb\u52a1<\/strong>\uff1a\u652f\u6301\u6570\u636e\u51c6\u5907\u548c\u8bad\u7ec3\u8fc7\u7a0b\uff0c\u5e2e\u52a9\u6a21\u578b\u5728\u5012\u8ba1\u65f6\u4efb\u52a1\u4e2d\u8fdb\u884c\u5b66\u4e60\u3002<\/li>\n<li><strong>\u4e58\u6cd5\u4efb\u52a1<\/strong>\uff1a\u652f\u6301\u6570\u636e\u51c6\u5907\u548c\u8bad\u7ec3\u8fc7\u7a0b\uff0c\u5e2e\u52a9\u6a21\u578b\u5728\u4e58\u6cd5\u4efb\u52a1\u4e2d\u8fdb\u884c\u5b66\u4e60\u3002<\/li>\n<li><strong>\u5355 GPU \u652f\u6301<\/strong>\uff1a\u9002\u7528\u4e8e\u6a21\u578b\u53c2\u6570\u5c0f\u4e8e\u7b49\u4e8e 1.5B \u7684\u60c5\u51b5\u3002<\/li>\n<li><strong>\u591a GPU \u652f\u6301<\/strong>\uff1a\u9002\u7528\u4e8e\u66f4\u5927\u53c2\u6570\u7684\u6a21\u578b\uff0c\u80fd\u591f\u5f00\u53d1\u590d\u6742\u7684\u63a8\u7406\u80fd\u529b\u3002<\/li>\n<li><strong>Instruct Ablation<\/strong>\uff1a\u652f\u6301 QWen-2.5-3B Instruct \u6a21\u578b\u7684\u5b9e\u9a8c\u3002<\/li>\n<li><strong>\u8d28\u91cf\u63d0\u5347\u5de5\u5177<\/strong>\uff1a\u5305\u62ec flash-attn\u3001wandb\u3001IPython \u548c matplotlib \u7b49\u5de5\u5177\uff0c\u63d0\u5347\u6a21\u578b\u8bad\u7ec3\u548c\u4f7f\u7528\u4f53\u9a8c\u3002<\/li>\n<\/ul>\n<p>&nbsp;<\/p>\n<h2>\u4f7f\u7528\u5e2e\u52a9<\/h2>\n<h3>\u5b89\u88c5\u6d41\u7a0b<\/h3>\n<ol>\n<li>\u521b\u5efa\u865a\u62df\u73af\u5883\uff1a\n<pre><code>conda create -n zero python=3.9\r\n<\/code><\/pre>\n<\/li>\n<li>\u5b89\u88c5 PyTorch\uff08\u53ef\u9009\uff09\uff1a\n<pre><code>pip install torch==2.4.0 --index-url https:\/\/download.pytorch.org\/whl\/cu121\r\n<\/code><\/pre>\n<\/li>\n<li>\u5b89\u88c5 vllm\uff1a\n<pre><code>pip3 install vllm==0.6.3\r\n<\/code><\/pre>\n<\/li>\n<li>\u5b89\u88c5 ray\uff1a\n<pre><code>pip3 install ray\r\n<\/code><\/pre>\n<\/li>\n<li>\u5b89\u88c5 verl\uff1a\n<pre><code>pip install -e .\r\n<\/code><\/pre>\n<\/li>\n<li>\u5b89\u88c5 flash-attn\uff1a\n<pre><code>pip3 install flash-attn --no-build-isolation\r\n<\/code><\/pre>\n<\/li>\n<li>\u5b89\u88c5\u8d28\u91cf\u63d0\u5347\u5de5\u5177\uff1a\n<pre><code>pip install wandb IPython matplotlib\r\n<\/code><\/pre>\n<\/li>\n<\/ol>\n<h3>\u529f\u80fd\u64cd\u4f5c\u6d41\u7a0b<\/h3>\n<h4>\u5012\u8ba1\u65f6\u4efb\u52a1<\/h4>\n<ol>\n<li>\u6570\u636e\u51c6\u5907\uff1a\n<pre><code>conda activate zero\r\npython .\/examples\/data_preprocess\/countdown.py --local_dir {path_to_your_dataset}\r\n<\/code><\/pre>\n<\/li>\n<li>\u8bad\u7ec3\u8fc7\u7a0b\uff1a\n<pre><code>conda activate zero\r\nexport N_GPUS=1\r\nexport BASE_MODEL={path_to_your_model}\r\nexport DATA_DIR={path_to_your_dataset}\r\nexport ROLLOUT_TP_SIZE=1\r\nexport EXPERIMENT_NAME=countdown-qwen2.5-0.5b\r\nexport VLLM_ATTENTION_BACKEND=XFORMERS\r\nbash .\/scripts\/train_tiny_zero.sh\r\n<\/code><\/pre>\n<\/li>\n<\/ol>\n<h4>3B+ \u6a21\u578b\u8bad\u7ec3<\/h4>\n<ol>\n<li>\u6570\u636e\u51c6\u5907\uff1a\n<pre><code>conda activate zero\r\npython examples\/data_preprocess\/countdown.py --template_type=qwen-instruct --local_dir={path_to_your_dataset}\r\n<\/code><\/pre>\n<\/li>\n<li>\u8bad\u7ec3\u8fc7\u7a0b\uff1a\n<pre><code>conda activate zero\r\nexport N_GPUS=2\r\nexport BASE_MODEL={path_to_your_model}\r\nexport DATA_DIR={path_to_your_dataset}\r\nexport ROLLOUT_TP_SIZE=2\r\nexport EXPERIMENT_NAME=countdown-qwen2.5-3b-instruct\r\nexport VLLM_ATTENTION_BACKEND=XFORMERS\r\nbash .\/scripts\/train_tiny_zero.sh\r\n<\/code><\/pre>\n<\/li>\n<\/ol>\n","protected":false},"excerpt":{"rendered":"<p>TinyZero \u662f\u4e00\u4e2a\u57fa\u4e8e veRL \u7684\u5f3a\u5316\u5b66\u4e60\u6a21\u578b\uff0c\u65e8\u5728\u590d\u73b0 DeepSeeK-R1 Zero \u5728\u5012\u8ba1\u65f6\u548c\u4e58\u6cd5\u4efb\u52a1\u4e2d\u7684\u8868\u73b0\u3002\u4ee4\u4eba\u60ca\u8bb6\u7684\u662f\uff0c\u8be5\u9879\u76ee\u4ec5\u9700 30 \u7f8e\u5143\u7684\u8fd0\u884c\u6210\u672c\uff08\u4f7f\u7528 2xH200\uff0c\u6bcf\u5c0f\u65f6 6.4 \u7f8e\u5143\uff0c\u4e0d\u5230 5 \u5c0f\u65f6\uff09\uff0c\u5c31\u80fd&#8230;<\/p>\n","protected":false},"author":1,"featured_media":61730,"comment_status":"open","ping_status":"closed","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[20],"tags":[230],"class_list":["post-19421","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-tool","tag-aikaiyuanxiangmu"],"_links":{"self":[{"href":"https:\/\/www.kdjingpai.com\/en\/wp-json\/wp\/v2\/posts\/19421","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.kdjingpai.com\/en\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.kdjingpai.com\/en\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.kdjingpai.com\/en\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.kdjingpai.com\/en\/wp-json\/wp\/v2\/comments?post=19421"}],"version-history":[{"count":0,"href":"https:\/\/www.kdjingpai.com\/en\/wp-json\/wp\/v2\/posts\/19421\/revisions"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/www.kdjingpai.com\/en\/wp-json\/wp\/v2\/media\/61730"}],"wp:attachment":[{"href":"https:\/\/www.kdjingpai.com\/en\/wp-json\/wp\/v2\/media?parent=19421"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.kdjingpai.com\/en\/wp-json\/wp\/v2\/categories?post=19421"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.kdjingpai.com\/en\/wp-json\/wp\/v2\/tags?post=19421"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}