From 89eab46e600c5381f44b1809b27263f0444b0104 Mon Sep 17 00:00:00 2001 From: BlinkDL Date: Wed, 11 Aug 2021 14:54:51 +0800 Subject: [PATCH] + info --- src/model.py | 4 ++++ train.py | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/src/model.py b/src/model.py index 35ea089..494ce5a 100644 --- a/src/model.py +++ b/src/model.py @@ -1,3 +1,7 @@ +######################################################################################################## +# The RWKV Language Model - https://github.com/BlinkDL/RWKV-LM +######################################################################################################## + import math import logging import numpy as np diff --git a/train.py b/train.py index 8374232..97ff024 100644 --- a/train.py +++ b/train.py @@ -1,3 +1,7 @@ +######################################################################################################## +# The RWKV Language Model - https://github.com/BlinkDL/RWKV-LM +######################################################################################################## + import os, sys, time, math, random, json, datetime import logging import numpy as np