如果科学计算任务可以分解成多个小任务，是选择 CPU 并行还是 GPU 并行呢？本篇大概讨论一下，做个记录，仅… Continue reading 科学计算任务CPU并行和GPU并行的选择
月度归档： 2024 年 6 月
贝里曲率为实数的证明
贝里曲率（Berry curvature）是量子力学和凝聚态物理中的一个重要概念，通常用于描述参数空间中的几何… Continue reading 贝里曲率为实数的证明
Python中的*args参数和**kwargs参数
在 Python 中，*args 和 **kwargs 是用来传递可变数量参数的机制，允许你编写灵活的函数。 … Continue reading Python中的*args参数和**kwargs参数
在薛定谔方程下自由粒子的解
本篇整理和复习下量子力学中的基础概念和推导，求解自由粒子的薛定谔方程。 对于一维情况，自由粒子的薛定谔方程为：… Continue reading 在薛定谔方程下自由粒子的解
当损失函数长时间不再下降时自动停止训练的设置
在机器学习的训练过程中，除了通过多次测试获取经验来设置总的训练轮次，也可以使用额外的代码进行判断，当损失函数长… Continue reading 当损失函数长时间不再下降时自动停止训练的设置
由于内存限制导致数据无法一次性从文件加载到PyTorch 中的简单解决方法
如果训练的数据比较大，无法一次性加载，简单的解决方法是分批从文件中加载。为了演示方便，这里的 x_train,… Continue reading 由于内存限制导致数据无法一次性从文件加载到PyTorch 中的简单解决方法
PyTorch张量数组的合并
这是之前的一篇：PyTorch基础（Tensor数据类型）。 在机器学习中，可能会涉及到多个数据合并后再进行训… Continue reading PyTorch张量数组的合并