Browse Source

不懂为什么

master
liupengfei 1 month ago
parent
commit
83689ccf6f
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      train.py

+ 2
- 0
train.py View File

@@ -169,6 +169,8 @@ def run_train():
config.training_dataset_class = tools.module_to_dict(datasets)[config.train_data]
flownet_train_gen = config.training_dataset_class(config.crop_type, config.crop_size, config.eval_size,
config.train_data_path)
print('rank fff =',rank)
print('group_size fff =',group_size)
sampler = datasets.DistributedSampler(flownet_train_gen, rank=rank, group_size=group_size, shuffle=True)
print('sampler')
train_dataset = ds.GeneratorDataset(flownet_train_gen, ["images", "flow"],


Loading…
Cancel
Save