Update README.md

README.md

@@ -16,6 +16,8 @@ Jam-so is a GPT2-like model for research in fine-grained Java analysis. It is in
 
 - We train the model on [training set](https://huggingface.co/datasets/apcl/so13m/blob/main/train.bin) for 1 epoch, roughly 300,000 training iterations.
 
+- Our [GitHub repo](https://github.com/apcl-research/jam/blob/main) contains the code for re-training using the [raw data](https://huggingface.co/datasets/apcl/so13m/blob/main/so13m.pkl).
+
 | Hyperparameter | Description | Value |
 | ----------- | ----------- |------------|
 |e | embedding dimensions              | 1024 |		 
@@ -28,7 +30,7 @@ Jam-so is a GPT2-like model for research in fine-grained Java analysis. It is in
 |r | learning rate                     | 3e-5 |		 
 |y | weight decay						| 1e-1 |	 
 
-We train our models using a single NVidia A5000 GPUs. Our [GitHub repo](https://github.com/apcl-research/jam/blob/main) contains the code for re-training using the [raw data](https://huggingface.co/datasets/apcl/so13m/blob/main/so13m.pkl).
+We train our models using a single NVidia A5000 GPUs. 
 
 ---
 ## Jam Projects