pytorch使用tips

一、数据加载
1.给collate_fn函数传参
(1)使用lambda函数

info = args.info	# info是已经定义过的
loader = Dataloader(collate_fn=lambda x: collate_fn(x, info))

这里巧用lambda函数, 相当于使用collate_fn函数再定义了一个匿名函数.

你可能感兴趣的:(工具类,python,pytorch)