【问题标题】:How to use a custom sampler in your Flux.jl data loader?如何在 Flux.jl 数据加载器中使用自定义采样器?
【发布时间】:2021-07-31 12:55:41
【问题描述】:

我正在尝试仅加载数据集的一部分,并尝试随机抽样等步骤,以了解这如何影响我的模型性能。我在这里阅读 Flux.jl 文档:https://fluxml.ai/Flux.jl/stable/data/dataloader/ 并尝试查看是否可以定义自己的自定义采样器,但它没有显示任何内容。是否可以定义这种采样器?

【问题讨论】:

    标签: julia flux.jl


    【解决方案1】:

    Dataloader 只是 Base.iterate 方法已被重载的类型

    @propagate_inbounds function Base.iterate(d::DataLoader, i=0)
    ...
    

    所以你应该能够做同样的事情,我想从 dataloader 类型继承并为你自己的类型重载 Base.iterate 并使用你想要的任何功能剪了它。只要确保返回一个包含 (batch, next_index) 的元组。有关更多信息,请参阅Flux.Dataloader 的源代码。

    【讨论】:

      猜你喜欢
      • 2020-09-03
      • 2022-12-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2021-11-09
      • 2021-12-29
      • 1970-01-01
      相关资源
      最近更新 更多