@icyfenix-dmla/cli 2026.5.2-2022 → 2026.5.2-2025
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/package.json +1 -1
- package/src/commands/data.js +2 -11
package/package.json
CHANGED
package/src/commands/data.js
CHANGED
|
@@ -401,19 +401,16 @@ async function downloadDatasets() {
|
|
|
401
401
|
// 构建选项列表
|
|
402
402
|
const choices = DATASETS.map((dataset, index) => {
|
|
403
403
|
const downloaded = isDatasetDownloaded(dataPath, dataset.id)
|
|
404
|
-
const isMnist = dataset.id === 'mnist'
|
|
405
404
|
|
|
406
405
|
let message = `${dataset.name} (${dataset.size})`
|
|
407
406
|
if (downloaded) {
|
|
408
407
|
message += ' [已下载]'
|
|
409
|
-
} else if (isMnist) {
|
|
410
|
-
message += ' [训练时自动下载]'
|
|
411
408
|
}
|
|
412
409
|
|
|
413
410
|
return {
|
|
414
411
|
name: index.toString(),
|
|
415
412
|
message,
|
|
416
|
-
disabled: downloaded
|
|
413
|
+
disabled: downloaded
|
|
417
414
|
}
|
|
418
415
|
})
|
|
419
416
|
|
|
@@ -428,7 +425,7 @@ async function downloadDatasets() {
|
|
|
428
425
|
message: '选择要下载的数据集',
|
|
429
426
|
choices,
|
|
430
427
|
hint: '空格选择，回车确认下载',
|
|
431
|
-
warn: '
|
|
428
|
+
warn: '已下载'
|
|
432
429
|
})
|
|
433
430
|
|
|
434
431
|
if (!selected || selected.length === 0) {
|
|
@@ -444,12 +441,6 @@ async function downloadDatasets() {
|
|
|
444
441
|
console.log()
|
|
445
442
|
console.log(chalk.cyan(`────────────────────────────────────`))
|
|
446
443
|
|
|
447
|
-
// MNIST 特殊处理
|
|
448
|
-
if (dataset.id === 'mnist') {
|
|
449
|
-
console.log(chalk.yellow('MNIST 数据集将在训练时通过 torchvision 自动下载'))
|
|
450
|
-
continue
|
|
451
|
-
}
|
|
452
|
-
|
|
453
444
|
// 检查是否已下载
|
|
454
445
|
if (isDatasetDownloaded(dataPath, dataset.id)) {
|
|
455
446
|
console.log(chalk.yellow(`${dataset.name} 已下载，跳过`))
|