Train from --data path/to/dataset.zip feature (#4185)

* Train from `--data path/to/dataset.zip` feature

* Update dataset_stats()

* cleanup

* cleanup2
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-07-28 02:04:10 +02:00
1 parent 3fef11706c
commit 5d66e48723
9 files changed
+122 -73

No files matched your search

+1 -3
View File
@@ -123,9 +123,7 @@ def run(data,
# model = nn.DataParallel(model)
# Data
with open(data, encoding='ascii', errors='ignore') as f:
data = yaml.safe_load(f)
check_dataset(data) # check
data = check_dataset(data) # check
# Half
half &= device.type != 'cpu' # half precision only supported on CUDA