diff --git a/dataset/dataset.py b/dataset/dataset.py
index b99c3a4..24c04e3 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -1,3 +1,13 @@
+"""
+Author: LiangSong(sl12160010@gmail.com)
+Date: 2023-04-24 20:05:21
+LastEditors: LiangSong(sl12160010@gmail.com)
+LastEditTime: 2023-04-24 20:05:59
+FilePath: /Open-Llama/dataset/dataset.py
+Description: 
+
+Copyright (c) 2023 by LiangSong(sl12160010@gmail.com), All Rights Reserved. 
+"""
 import torch
 import random
 from glob import glob
diff --git a/pretrain.py b/pretrain.py
index 18957d1..231566c 100644
--- a/pretrain.py
+++ b/pretrain.py
@@ -2,7 +2,7 @@
 Author: LiangSong(sl12160010@gmail.com)
 Date: 2023-04-12 19:12:42
 LastEditors: LiangSong(sl12160010@gmail.com)
-LastEditTime: 2023-04-12 21:01:32
+LastEditTime: 2023-04-24 20:06:19
 FilePath: /Open-Llama/pretrain.py
 Description: 
 
diff --git a/solver/trainer.py b/solver/trainer.py
index 878eba3..f9f3be8 100644
--- a/solver/trainer.py
+++ b/solver/trainer.py
@@ -1,3 +1,13 @@
+"""
+Author: LiangSong(sl12160010@gmail.com)
+Date: 2023-04-24 20:05:21
+LastEditors: LiangSong(sl12160010@gmail.com)
+LastEditTime: 2023-04-24 20:06:07
+FilePath: /Open-Llama/solver/trainer.py
+Description: 
+
+Copyright (c) 2023 by LiangSong(sl12160010@gmail.com), All Rights Reserved. 
+"""
 import os
 import time
 import wandb