定义loss_show类and 权重初始化init_weight_pytorch

weixin_44576543

于 2021-11-22 11:38:42 发布

阅读量1.5k

点赞数 1

分类专栏： python 文章标签： python

本文链接：https://blog.youkuaiyun.com/weixin_44576543/article/details/121468288

版权

本文介绍了如何在PyTorch中定义一个名为losshistory的损失类，并展示了如何调用该类以及权重初始化函数init_weight_pytorch的使用过程。

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

1.define the class :losshistory 定义loss类

import os
import numpy as np
import scipy.signal
from matplotlib import pyplot as plt

class LossHistory():
    def __init__(self, log_dir):
        import datetime
        curr_time = datetime.datetime.now()
        time_str = datetime.datetime.strftime(curr_time,'%Y_%m_%d_%H_%M_%S')
        self.log_dir    = log_dir
        self.time_str   = time_str
        self.save_path  = os.path.join(self.log_dir, "loss_" + str(self.time_str))
        self.acc        = []
        self.losses     = []
        self.val_loss   = []
        
        os.makedirs(self.save_path)

    def append_loss(self, acc, loss, val_loss):
        self.acc.append(acc)
        self.losses.append(loss)
        self.val_loss.append(val_loss)
        with open(os.path.join(self.save_path, "epoch_acc_" + str(self.time_str) + ".txt"), 'a') as f:

最低0.47元/天解锁文章