2023年10月8日
pytorch训练模版
摘要： train.py import torch import numpy as np import os import math import sys import argparse import psutil import torch import torch.optim as optim impor 阅读全文
posted @ 2023-10-08 15:27 心比天高xzh 阅读(80) 评论(0) 推荐(0)
python命令行传参
摘要： #run.py import argparse parser = argparse.ArgumentParser() parser.add_argument('--device', type=str, default='cpu:0',help='use which device') parser.a 阅读全文
posted @ 2023-10-08 14:50 心比天高xzh 阅读(26) 评论(0) 推荐(0)
pytorch 自定义dataset类
摘要： 实现模版 class our_dataset(Dataset): def __init__(self,···): super(our_dataset, self).__init__() #初始化，可以自定义添加参数 def __getitem__(self,index): ··· return im 阅读全文
posted @ 2023-10-08 14:29 心比天高xzh 阅读(36) 评论(0) 推荐(0)
python查看占用系统内存
摘要： import os import psutil info = psutil.virtual_memory() print(u'cpu个数：',psutil.cpu_count()) print(u'电脑总内存：%.4f GB' % (info.total / 1024 / 1024 / 1024)) 阅读全文
posted @ 2023-10-08 14:22 心比天高xzh 阅读(41) 评论(0) 推荐(0)