Update README.md

README.md

@@ -13,6 +13,8 @@ repo: N.A.
 
 <!-- Provide a quick summary of what the model is/does. -->
 
+This model is trained as part of the coursework of COMP34812.
+
 This is a binary classification model that was trained with prompt input to
       detect whether two pieces of text were written by the same author.
 
@@ -23,8 +25,8 @@ This is a binary classification model that was trained with prompt input to
 
 <!-- Provide a longer summary of what this model is. -->
 
-This model is based upon a Llama2 model that was fine-tuned
-      on 30K pairs of texts for authorship verification. The model is trained with prompt inputs to utilize the model's linguistic knowledge.
+This model is based on a Llama2 model that was fine-tuned
+      on 30K pairs of texts for authorship verification. The model is fine-tuned with prompt inputs to utilize the model's linguistic knowledge.
       To run the model, the demo code is provided in demo.ipynb submitted.
       It is advised to use the pre-processing and post-processing functions (provided in demo.ipynb) along with the model for best results.
 
@@ -78,7 +80,7 @@ This model is based upon a Llama2 model that was fine-tuned
 
       - trained on: V100 16GB
       - overall training time: 59 hours
-      - duration per training epoch: 59 minutes
+      - duration per training epoch: 59 hours
       - model size: ~27GB
       - LoRA adaptor size: 192 MB
 
@@ -125,8 +127,10 @@ The development set provided, amounting to 6K pairs.
 ### Software
 
 
-      - Transformers 4.18.0
-      - Pytorch 1.11.0+cu113
+      - Transformers
+      - Pytorch
+      - bitesandbytes
+      - Accelerate
 
 ## Bias, Risks, and Limitations
 
@@ -135,10 +139,3 @@ The development set provided, amounting to 6K pairs.
 Any inputs (concatenation of two sequences plus prompt words) longer than
       4096 subwords will be truncated by the model.
 
-## Additional Information
-
-<!-- Any other information that would be useful for other people to know. -->
-
-The hyperparameters were determined by experimentation
-      with different values, such that the model could succesfully train on the V100 with a gradual decrease in training loss. Since LoRA is used, the Llama2 base model must also
-      be loaded for the model to function, pre-trained Llama2 model access would need to be requested, access could be applied on https://huggingface.co/meta-llama.