Q:如何简单的构建一个batch?
data_collator(eval_dataset[:3])
这种操作是不ok的。
有一些代码喜欢把数据的预处理放在 data_collator 函数内,有时候我们希望自己构建一个batch简单的测试一下。
下述是构建batch的代码
batch = [eval_dataset[i] for i in range(3)]
data = data_collator(batch)
print(data)
Q:如何简单的构建一个batch?
data_collator(eval_dataset[:3])
这种操作是不ok的。
有一些代码喜欢把数据的预处理放在 data_collator 函数内,有时候我们希望自己构建一个batch简单的测试一下。
下述是构建batch的代码
batch = [eval_dataset[i] for i in range(3)]
data = data_collator(batch)
print(data)