PyTorch 中如何调整训练数据集大小（--trainsize 参数）

如果训练自己的输入数据集大小为 2002003，你可以将参数 --trainsize 设置为 200。修改后的代码如下：

parser.add_argument('--trainsize', type=int, default=200, help='training dataset size')

这样设置后，训练数据集的大小就会被设置为 200。注意，这个值只是指定了训练数据集的大小，实际数据集的大小仍然需要根据你的数据集进行调整。