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:
1 parent
3fef11706c
commit
5d66e48723
9 files changed
+122
-73
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user