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