个人简介:研一在读,目前方向是talking face generation, 欢迎来交流~
while data['imgs'][0].shape[0] % batch_size ==0: batch_size = batch_size-1 num_batch = data['imgs'][0].shape[0] // batch_size + 1
vico_baseline在提取3d 参数时有128的bug , num_batch = data['imgs'][0].shape[0] // batch_size + 1, 在这句后面
sudo pip install --upgrade youtube_dl 保证安装的是最新版
torch.hub可以快捷调用 ``` import torch model = torch.hub.load('pytorch/vision:v0.10.0', 'deeplabv3_resnet50', pretrained=True) model.eval() ```
workbench.editor.enablePreview 设置搜索这个, 把勾去掉, 可以让vscode每次默认打开新标签页
a, b, and c 萌姐说是这样的
Segmentation fault (core dumped) 的错误原来是因为我没安cudnn引起的
echo 3 > /proc/sys/vm/drop_caches 用来暂时解决Segmentation fault (core dumped)的问题
https://blog.51cto.com/u_15316394/3215728, 多进程数据共享, 包括list等
https://www.gairuo.com/p/python-multiprocessing, 很全的关于python多进程的解释
/etc/bash.bashrc 更改了, 参见https://blog.csdn.net/weixin_44384491/article/details/121142093
tem = torch.ones(gt_pose.shape).cuda() pred_just_pose = torch.where(pred_just_pose<1.0,pred_just_pose,tem) gt_pose = torch.where(gt_pose<1.0,gt_pose, tem)
b = torch.where(a == 1, torch.Tensor([-1]), a) torch.where用法
torch.set_printoptions(precision=4,sci_mode=False) 打印tensor为小数格式
fuser -v /dev/nvidia0 残存的没有进程号的程序可以用这个命令查看
ps -ef |grep xx grep命令是查找, 可以查找用户或者python程序
pytorch训练结果的复现设置随机种⼦ def seed_torch(seed=2018): random.seed(seed) os.environ['PYTHONHASHSEED'] = str(seed) np.random.seed(seed) torch.manual_seed(seed) torch.cuda.manual_seed(seed) torch.backends.cudnn.deterministic = True