unitmath 训练笔记

AI算法网奇

于 2024-05-31 01:05:36 发布

阅读量84

点赞数

分类专栏：深度学习宝典文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/jacke121/article/details/139337947

版权

深度学习宝典专栏收录该内容

1027 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文记录了使用unitmath在Windows环境下进行深度学习模型的多GPU训练过程，主要涉及pytorch框架的相关操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

unitmath windows训练代码：多卡

import argparse
import logging
import os
import pprint

import torch
from torch import nn
import torch.backends.cudnn as cudnn
from torch.optim import SGD
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter
import yaml

from dataset.semi import SemiDataset
from model.semseg.deeplabv3plus import DeepLabV3Plus
from supervised import evaluate
from util.classes import CLASSES
from util.ohem import ProbOhemCrossEntropy2d
from util.utils import count_params, init_log, AverageMeter
from util.dist_helper import setup_distributed


parser = argparse.ArgumentParser(description='Revisiting Weak-to-Strong Consistency in Semi-Supervised Semantic Segmentation'