PyTorch 框架
PyTorch简洁易上手,便于科研。
JNingWei
工作后比较忙,不怎么看账号和消息。回复不及时望见谅。
展开
-
【pytorch】ncnn
onnx转ncnn完整实现:def onnx2ncnn(self): assert os.path.isfile(simplified_onnx_path) os.system('onnx2ncnn {} {} {}'.format(simplified_onnx_path, param_path, bin_path)) print('\n param has been save to {}'.format(param_path)) print(' bin has原创 2021-09-22 01:03:19 · 307 阅读 · 0 评论 -
【pytorch】tensorboardX
关于自定义runs下为什么不能正常保存tb记录(一直为空)的原因,可以尝试把 SummaryWriter 的如下参数实时打印出来:self.logdir = logdirself.purge_step = purge_stepself._max_queue = max_queueself._flush_secs = flush_secsself._filename_suffix = filename_suffixself._write_to_disk = write_to_diskself.k原创 2021-09-22 00:52:51 · 142 阅读 · 0 评论 -
【pytorch】tnn
首先确保已经安装了tnn环境:onnx2tnn转tnn:def onnx2tnn(self): assert os.path.isfile(simplified_onnx_path) os.system('cd {}\n' 'python onnx2tnn.py {} -version=v1.0 -optimize=1 -half=1 -o {} -input_shape input:1,{},{},{}' .format(C.o原创 2021-09-22 00:49:08 · 1053 阅读 · 0 评论 -
【pytorch】dropout
在pytorch中,nn.Dropout(xx) 中的参数表示要丢弃的比例,和tensorflow的定义是反过来的。原创 2021-09-22 00:46:50 · 185 阅读 · 0 评论 -
【pytorch】onnx
pytorch任意形式的model(.t7、.pth等等)转.onnx全都可以采用固定格式:model_dir = './'pth_path = model_dir + 'A.pth'onnx_path = model_dir + 'A.onnx'batch_size = 1input_shape = (3, 112, 112)cfg = Config()cfg.load_from_file(args.model_cfg_file)model = PFLD_SE3_eval(cfg.mo原创 2021-09-22 00:45:26 · 369 阅读 · 0 评论 -
【pytorch】bn
bn接口定义:torch.nn.BatchNorm2d:def init(self,num_features,eps=1e-5,momentum=0.1,affine=True,track_running_stats=True)args:momentum:默认为 0.1 。要freeze的时候就设置为0.0(和 tf 里面是反着来的,tf是设置为1.0才能freeze)。rack_running_stats:计算running_mean和running_var(即mo原创 2021-09-22 00:41:14 · 663 阅读 · 0 评论 -
【pytorch】model
打印网络结构(不带节点名称):for ele in model.modules(): print(ele)打印named_parameters():for (name, param) in model.named_parameters(): if not param.requires_grad: print(name, param.data)几个主要api的区别:pytorch model.named_parameters() ,model.parameters(原创 2021-09-22 00:37:41 · 301 阅读 · 0 评论 -
【pytorch】freeze
freeze bn:把所有相关的bn设置为 momentum=1.0 。freeze 正常参数:先比较两个state_dict,来freeze交集:def freeze_model(model, defined_dict, keep_step=None): for (name, param) in model.named_parameters(): if name in defined_dict: param.requires_grad = Fals原创 2021-09-22 00:34:33 · 514 阅读 · 0 评论 -
【pytorch】tensor
import torcha = torch.tensor([1, 2, 3])print(a) # tensor([1, 2, 3])print(a.item()) # 会报错,因为 item() 只适配输出单个变量,无法适配list或者np.ndarrayprint(a.numpy()) # [1, 2, 3]原创 2021-09-22 00:30:11 · 135 阅读 · 0 评论 -
【pytorch】lr
scheduler唯一作用就是 改optimizer的lr 。所以scheduler可以弃用,自己写个函数来实时 optimizer.param_groups[0][‘lr’] = lr 即可。因此,虽然print lr有 optimizer.param_groups[0][‘lr’] 和 scheduler.get_lr()[0] 这两种形式,但是前者才是正道,后者只是借用scheduler调控optimizer的lr时,才附加送的便捷方式。...原创 2021-09-22 00:29:21 · 218 阅读 · 0 评论 -
【pytorch】固定(freeze)住部分网络
【pytorch】固定(freeze)住部分网络原创 2021-09-15 00:44:44 · 16359 阅读 · 5 评论 -
【pytorch】安装pytorch
安装通过condaconda install pytorch torchvision -c pytorch通过pippip install torch torchvision原创 2020-03-15 15:40:44 · 650 阅读 · 0 评论 -
pytorch: tensor类型的构建与相互转换
Summary主要包括以下三种途径:使用独立的函数;使用torch.type()函数;使用type_as(tesnor)将张量转换为给定类型的张量。使用独立函数import torchtensor = torch.randn(3, 5)print(tensor)# torch.long() 将tensor投射为long类型long_tensor = te...原创 2018-04-08 11:33:30 · 106690 阅读 · 1 评论 -
pytorch: PyTorch中 使用 Tensorboard
介绍因为工作需要,改用pytorch。但如何将训练过程可视化成了大问题。听说pytorch代码中可以插入tensorboard代码,第一反应是居然可以这么玩。。网络上PyTorch中使用tensorboard的方法有很多。但毕竟tensorboard不是PyTorch框架原生自带的,因此大多方法都只能支持部分功能。经过孙大佬的推荐,觉得使用tensorboardX应该是目前已知的最好方法了...原创 2018-03-29 12:53:14 · 21358 阅读 · 4 评论 -
pytorch: 框架介绍
Tutorialpytorch.org/tutorials莫烦Python-Pytorch教程系列Orgpytorch.orgCommunitygithub.com/pytorchModelgithub.com/pytorch/visionSource Codegithub.com/pytorch/pytorchAdvantages支持动态图更易于研究原创 2018-01-10 18:45:10 · 1686 阅读 · 0 评论