Initial

README.md

@@ -51,6 +51,13 @@ $ cd VEGA_AE
 
 - Install Python (Version 3.8.1) in Conda environment.
 
+We provide a pre-packaged conda virtual environment in ```./vega_ae.yml```, which includes specific versions of Python and required extension packages. The python environment can be directly created using the following command.
+
+```
+conda env create -f vega_ae.yml
+```
+
+The python environment can also be created using the following command.
 ```
 $ conda create -n vega_ae python=3.8.1
 $ conda activate vega_ae
@@ -114,7 +121,7 @@ Customize parameters for code generation by modifying following options in the `
  --model_name_or_path ../../models/UnixCoder \
  --test_filename ../../dataset/test.jsonl \
  --output_dir ../../models/FT_Model \
- --beam_size 4 \
+ --beam_size 1 \
  --train_batch_size 256 \
  --eval_batch_size 256 \
  --learning_rate 6e-5 \

run_function_test.sh

@@ -5,7 +5,7 @@ python ./Scripts/UnixCoder/run_one_model.py \
  --model_name_or_path ../../models/UnixCoder \
  --test_filename ../../dataset/test.jsonl \
  --output_dir ../../models/FT_Model \
- --beam_size 4 \
+ --beam_size 1 \
  --train_batch_size 256 \
  --eval_batch_size 256 \
  --learning_rate 6e-5 \

run_test.sh

@@ -5,7 +5,7 @@ python ./Scripts/UnixCoder/run_one_model.py \
  --model_name_or_path ../../models/UnixCoder \
  --test_filename ../../dataset/test.jsonl \
  --output_dir ../../models/FT_Model \
- --beam_size 4 \
+ --beam_size 1 \
  --train_batch_size 256 \
  --eval_batch_size 256 \
  --learning_rate 6e-5 \

vega_ae.yml

@@ -0,0 +1,81 @@
+name: vega_ae
+channels:
+  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
+  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
+  - defaults
+dependencies:
+  - _libgcc_mutex=0.1=main
+  - _openmp_mutex=5.1=1_gnu
+  - ca-certificates=2024.9.24=h06a4308_0
+  - ld_impl_linux-64=2.40=h12ee557_0
+  - libedit=3.1.20230828=h5eee18b_0
+  - libffi=3.2.1=1
+  - libgcc-ng=11.2.0=h1234567_1
+  - libgomp=11.2.0=h1234567_1
+  - libstdcxx-ng=11.2.0=h1234567_1
+  - ncurses=6.4=h6a678d5_0
+  - openssl=1.1.1w=h7f8727e_0
+  - pip=24.2=py38h06a4308_0
+  - python=3.8.1=h0371630_1
+  - readline=7.0=h7b6447c_5
+  - setuptools=75.1.0=py38h06a4308_0
+  - sqlite=3.33.0=h62c20be_0
+  - tk=8.6.11=h1ccaba5_0
+  - wheel=0.44.0=py38h06a4308_0
+  - xz=5.4.6=h5eee18b_1
+  - zlib=1.2.11=0
+  - pip:
+    - certifi==2024.8.30
+    - charset-normalizer==3.4.0
+    - cmake==3.30.5
+    - filelock==3.16.1
+    - fsspec==2024.10.0
+    - fuzzywuzzy==0.18.0
+    - huggingface-hub==0.26.2
+    - idna==3.10
+    - jinja2==3.1.4
+    - levenshtein==0.25.1
+    - lit==18.1.8
+    - markupsafe==2.1.5
+    - mpmath==1.3.0
+    - networkx==3.1
+    - numpy==1.24.4
+    - nvidia-cublas-cu11==11.10.3.66
+    - nvidia-cublas-cu12==12.1.3.1
+    - nvidia-cuda-cupti-cu11==11.7.101
+    - nvidia-cuda-cupti-cu12==12.1.105
+    - nvidia-cuda-nvrtc-cu11==11.7.99
+    - nvidia-cuda-nvrtc-cu12==12.1.105
+    - nvidia-cuda-runtime-cu11==11.7.99
+    - nvidia-cuda-runtime-cu12==12.1.105
+    - nvidia-cudnn-cu11==8.5.0.96
+    - nvidia-cudnn-cu12==8.9.2.26
+    - nvidia-cufft-cu11==10.9.0.58
+    - nvidia-cufft-cu12==11.0.2.54
+    - nvidia-curand-cu11==10.2.10.91
+    - nvidia-curand-cu12==10.3.2.106
+    - nvidia-cusolver-cu11==11.4.0.1
+    - nvidia-cusolver-cu12==11.4.5.107
+    - nvidia-cusparse-cu11==11.7.4.91
+    - nvidia-cusparse-cu12==12.1.0.106
+    - nvidia-nccl-cu11==2.14.3
+    - nvidia-nccl-cu12==2.19.3
+    - nvidia-nvjitlink-cu12==12.6.77
+    - nvidia-nvtx-cu11==11.7.91
+    - nvidia-nvtx-cu12==12.1.105
+    - packaging==24.1
+    - pathlib==1.0.1
+    - python-levenshtein==0.25.1
+    - pyyaml==6.0.2
+    - rapidfuzz==3.9.7
+    - regex==2024.9.11
+    - requests==2.32.3
+    - sympy==1.13.3
+    - tokenizers==0.13.3
+    - torch==2.0.1
+    - tqdm==4.66.2
+    - transformers==4.28.0
+    - triton==2.0.0
+    - typing-extensions==4.12.2
+    - urllib3==2.2.3
+prefix: /home/ict_zhongm/.conda/envs/vega_ae