trainer/train_test.py

# -*- coding: utf-8 -*-
"""
Author: Kungia
@Software: Pycharm
@file: train_test.py
@Author: 28963
@dateTime: 2025/8/5 17:22
@brief: 
"""


# import torch
# # print(torch.cuda.is_available())
# print(torch.version.cuda)