DeepWok / mase

Machine-Learning Accelerator System Exploration Tools
Other
117 stars 52 forks source link

Fix Bugs in Lab Notebook and CLI Transform Command #12

Closed bet20ICL closed 8 months ago

bet20ICL commented 8 months ago
  File "/data/bet20/mase/machop/chop/cli.py", line 350, in _run_transform
    transform(**transform_params)
  File "/data/bet20/mase/machop/chop/actions/transform.py", line 70, in transform
    graph, _ = add_common_metadata_analysis_pass(graph, pass_args={"dummy_in": dummy_in})
  File "/data/bet20/mase/machop/chop/passes/graph/analysis/add_metadata/add_common_metadata.py", line 382, in add_common_metadata_analysis_pass
    graph = graph_iterator_for_metadata(graph, **pass_args)
  File "/data/bet20/mase/machop/chop/passes/graph/analysis/add_metadata/add_common_metadata.py", line 203, in graph_iterator_for_metadata
    result = modules[node.target](*args, **kwargs)
  File "/home/bet20/miniconda3/envs/mase-adls/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/home/bet20/miniconda3/envs/mase-adls/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/bet20/miniconda3/envs/mase-adls/lib/python3.10/site-packages/torch/nn/modules/batchnorm.py", line 171, in forward
    return F.batch_norm(
  File "/home/bet20/miniconda3/envs/mase-adls/lib/python3.10/site-packages/torch/nn/functional.py", line 2476, in batch_norm
    _verify_batch_size(input.size())
  File "/home/bet20/miniconda3/envs/mase-adls/lib/python3.10/site-packages/torch/nn/functional.py", line 2444, in _verify_batch_size
    raise ValueError(f"Expected more than 1 value per channel when training, got input size {size}")
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 16])
Aaron-Zhao123 commented 8 months ago

Can you try use a larger batch_size?

Aaron-Zhao123 commented 8 months ago

Refactored in PR #15