Update README.md

README.md

@@ -15,6 +15,8 @@ Jam is a GPT2-like model for research in fine-grained Java analysis. It is inten
 - The dataset used to train our model is our own dataset [jm52m dataset](https://huggingface.co/datasets/apcl/jm52m), which consists of the processed source code of 52 million Java methods. 
 
 - We train the model on [training set](https://huggingface.co/datasets/apcl/jm52m/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/jm52m/blob/main/fundats-j1.pkl)
 
 | Hyperparameter | Description | Value |
 | ----------- | ----------- |------------|
@@ -28,7 +30,7 @@ Jam is a GPT2-like model for research in fine-grained Java analysis. It is inten
 |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/jm52m/blob/main/fundats-j1.pkl)
+We train our models using a single NVidia A5000 GPUs. 
 
 ---
 ## Jam Projects