1. 初始化代码;
This commit is contained in:
1
.python-version
Normal file
1
.python-version
Normal file
@@ -0,0 +1 @@
|
|||||||
|
3.11
|
||||||
10
main.py
Normal file
10
main.py
Normal file
@@ -0,0 +1,10 @@
|
|||||||
|
import logging.config
|
||||||
|
logging.config.fileConfig('./confs/logging.conf')
|
||||||
|
|
||||||
|
from models.transformer import Transformer
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
logging.info("Running Transformer Model")
|
||||||
|
model_runner = Transformer({})
|
||||||
|
model_runner.run_test()
|
||||||
12
pyproject.toml
Normal file
12
pyproject.toml
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
[project]
|
||||||
|
name = "nlp-learning"
|
||||||
|
version = "0.1.0"
|
||||||
|
description = "Add your description here"
|
||||||
|
readme = "README.md"
|
||||||
|
requires-python = ">=3.11"
|
||||||
|
dependencies = [
|
||||||
|
"matplotlib>=3.10.8",
|
||||||
|
"seaborn>=0.13.2",
|
||||||
|
"torch>=2.9.1",
|
||||||
|
"torchvision>=0.24.1",
|
||||||
|
]
|
||||||
Reference in New Issue
Block a user