Files
composable_kernel/tile_engine/ops/grouped_conv/problems/validation_holdout.py
Yaswanth Raparti 6989cf800c [rocm-libraries] ROCm/rocm-libraries#6327 (commit 1e7a12e)
[CK][CK TILE] Dispatcher kernel selection heuristic for
 grouped conv (#6327)

## Motivation
The ML heuristic in dispatcher does not support grouped-conv operator
yet. In this PR, the support for fwd, bdw-data, and bwd-weight
grouped-conv kernels have been added. A tile_engine utility has also
been added to compile and run any selected kernel configuration through
dispatcher infrastructure.

## Technical Details

1. Tile engine utility is added to benchmark each shape with all the
possible kernel+tile_size combinations here -
[https://github.com/ROCm/rocm-libraries/blob/users/yraparti/ck/dispatcher-grouped-conv-heuristics/projects/composablekernel/tile_engine/ops/grouped_conv/grouped_conv_full_benchmark.py](url)
2. New LGBM regressor models for grouped conv are added to models
directory. We have 3 separate models for fwd, bwd-data, and bwd-weights
[https://github.com/ROCm/rocm-libraries/tree/users/yraparti/ck/dispatcher-grouped-conv-heuristics/projects/composablekernel/dispatcher/heuristics/models](url)
3. Implemented lazy GPU initialization (dispatcher/python)
- **Issue**: ProcessPoolExecutor fork() + GPU context caused memory
access faults
- **Solution**: Mirror FMHA pattern - defer GPU initialization until
first run()
  - **Changes**:
- setup_multiple_grouped_conv_dispatchers() returns List[Path], not
loaded libs
    - GpuGroupedConvRunner.__init__() no longer calls ctypes.CDLL
    - Added _ensure_initialized() method for lazy GPU loading
    - GPU context created only on first run() call
  - **Benefit**: Parallel compilation now works without GPU conflicts
4. Addressed few miscellaneous issues such as:
  - Fixed BF16->FP16 naming bug in the dispatcher wrapper
- Added new tile sizes, and comp_v5 pipeline to the arch spec to expand
the kernel selection
- Added automatic padding support for unsupported shapes in dispatcher
runner
- Created a single source of truth between tile_engine and dispatcher
about the architecture and tile_size details
- Build a validation scripts to compare oracle_best vs ml_heuristic
comparison

## Test Plan

1. Validated fwd, bwd-data, and bwd-weight kernels with both known and
unseen data sets with up to 300 problems.
2. Ensured that test cases are added in both dispatcher and tile_engine
to validate the heuristic.

## Test Result
Results on Unseen shapes validated on gfx950
#### Forward Pass Model
- **Training Data**: 48,845 measurements across 1,372 unique problem
shapes
- **Validation Set**: 300 unseen problems from model crawler
- **Validation Performance** (vs. oracle):
  - Mean Efficiency: **93.05%**
  - Median Efficiency: **96.8%**
  - P10 Efficiency: **79.9%**

#### Backward Data Gradient (bwd_data) Model
- **Training Data**: 18,773 measurements across 891 unique problem
shapes
- **Validation Set**: 300 unseen problems from model crawler
- **Validation Performance** (vs. oracle):
  - Mean Efficiency: **93.8%**
  - Median Efficiency: **96.5%**
  - P10 Efficiency: **82.9%**

#### Backward Weight Gradient (bwd_weight) Model
- **Training Data**: 34,900 measurements across 1,508 unique problem
shapes
- **Validation Set**: 300 unseen problems from model crawler
- **Validation Performance** (vs. oracle):
  - Mean Efficiency: **96.1%**
  - Median Efficiency: **99.2%**
  - P10 Efficiency: **89.4%**

## Submission Checklist

- [ x] Look over the contributing guidelines at
https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.
2026-05-08 20:48:42 +00:00

2410 lines
71 KiB
Python

"""
Validation holdout set for heuristic testing.
300 problems (250 2D + 50 3D) randomly sampled for validation.
"""
from grouped_conv_utils import GroupedConvProblem
VALIDATION_PROBLEMS = [
GroupedConvProblem(
N=4, C=256, K=256, G=4,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=1024, K=256, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=64, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=128, G=1,
Di=1, Hi=112, Wi=112,
Z=1, Y=3, X=3,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=64, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=64, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=1024, K=128, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=1024, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=512, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=1024, K=64, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=64, K=64, G=8,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=128, G=1,
Di=1, Hi=112, Wi=112,
Z=1, Y=3, X=3,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=256, K=512, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=256, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=64, K=64, G=2,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=128, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=128, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=512, K=1024, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=1024, K=64, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=1024, K=512, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=256, K=64, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=32, K=32, G=2,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=64, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=128, C=128, K=128, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=512, K=128, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=64, G=64,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=128, G=4,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=512, K=512, G=8,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=128, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=1024, K=128, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=256, K=512, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=64, G=2,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=256, K=64, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=256, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=4, pad_w=4,
dilation_d=1, dilation_h=4, dilation_w=4
),
GroupedConvProblem(
N=4, C=64, K=128, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=256, K=256, G=1,
Di=1, Hi=14, Wi=14,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=1, C=128, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=4, pad_w=4,
dilation_d=1, dilation_h=4, dilation_w=4
),
GroupedConvProblem(
N=16, C=128, K=512, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=256, G=8,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=128, K=128, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=128, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=1024, K=64, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=32, C=128, K=64, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=512, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=128, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=1024, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=256, K=128, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=256, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=64, G=2,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=4, pad_w=4,
dilation_d=1, dilation_h=4, dilation_w=4
),
GroupedConvProblem(
N=32, C=128, K=512, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=512, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=128, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=512, K=128, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=512, K=128, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=64, K=128, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=512, K=64, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=6, pad_w=6,
dilation_d=1, dilation_h=6, dilation_w=6
),
GroupedConvProblem(
N=8, C=32, K=32, G=4,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=128, G=4,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=256, K=256, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=128, G=2,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=512, K=64, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=64, G=64,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=1024, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=1024, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=128, G=8,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=1024, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=4, C=256, K=512, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=512, K=128, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=1024, K=64, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=64, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=128, G=4,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=256, K=512, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=6, pad_w=6,
dilation_d=1, dilation_h=6, dilation_w=6
),
GroupedConvProblem(
N=16, C=256, K=256, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=1024, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=256, K=512, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=256, K=64, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=512, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=512, K=512, G=8,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=128, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=512, K=512, G=8,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=256, K=256, G=8,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=64, G=4,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=256, K=1024, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=128, G=2,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=128, G=4,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=256, K=128, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=64, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=1024, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=512, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=512, K=128, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=512, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=256, K=256, G=4,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=512, K=512, G=8,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=256, K=512, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=1024, K=64, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=512, K=512, G=8,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=256, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=128, K=128, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=16, C=256, K=512, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=8, C=256, K=128, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=1024, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=6, pad_w=6,
dilation_d=1, dilation_h=6, dilation_w=6
),
GroupedConvProblem(
N=8, C=256, K=512, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=4, pad_w=4,
dilation_d=1, dilation_h=4, dilation_w=4
),
GroupedConvProblem(
N=16, C=64, K=64, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=64, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=128, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=256, K=512, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=512, K=1024, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=16, K=16, G=2,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=128, G=128,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=64, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=512, K=1024, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=256, K=256, G=4,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=64, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=32, K=32, G=4,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=64, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=1, Hi=112, Wi=112,
Z=1, Y=1, X=1,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=256, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=128, K=128, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=128, K=256, G=1,
Di=1, Hi=112, Wi=112,
Z=1, Y=1, X=1,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=32, K=32, G=2,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=128, K=512, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=32, K=32, G=2,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=64, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=256, K=256, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=1024, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=256, K=128, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=64, G=1,
Di=1, Hi=112, Wi=112,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=512, K=512, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=512, G=1,
Di=1, Hi=14, Wi=14,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=16, C=128, K=1024, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=1024, K=256, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=32, K=32, G=4,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=128, K=512, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=4, pad_w=4,
dilation_d=1, dilation_h=4, dilation_w=4
),
GroupedConvProblem(
N=16, C=128, K=128, G=8,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=16, K=16, G=2,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=256, K=512, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=256, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=512, K=256, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=16, K=16, G=2,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=512, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=512, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=512, K=1024, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=256, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=512, K=512, G=512,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=1024, K=512, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=256, G=8,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=256, K=512, G=1,
Di=1, Hi=112, Wi=112,
Z=1, Y=1, X=1,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=256, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=16, C=256, K=256, G=4,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=512, K=64, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=256, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=512, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=64, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=64, K=64, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=64, G=64,
Di=1, Hi=112, Wi=112,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=256, G=256,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=128, K=128, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=64, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=128, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=8, C=256, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=128, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=1024, K=128, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=256, K=256, G=4,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=512, K=128, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=256, G=4,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=512, K=64, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=64, K=128, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=1, C=128, K=1024, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=512, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=256, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=4, C=128, K=512, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=64, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=32, K=32, G=2,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=256, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=256, G=1,
Di=1, Hi=112, Wi=112,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=64, K=512, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=1024, K=256, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=256, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=6, pad_w=6,
dilation_d=1, dilation_h=6, dilation_w=6
),
GroupedConvProblem(
N=32, C=512, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=64, K=128, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=256, K=128, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=128, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=512, K=256, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=128, G=1,
Di=1, Hi=112, Wi=112,
Z=1, Y=7, X=7,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=3, pad_w=3,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=128, G=1,
Di=1, Hi=14, Wi=14,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=6, pad_w=6,
dilation_d=1, dilation_h=6, dilation_w=6
),
GroupedConvProblem(
N=1, C=256, K=1024, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=256, K=256, G=1,
Di=1, Hi=14, Wi=14,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=16, C=128, K=512, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=256, K=512, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=16, C=64, K=128, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=64, G=8,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=128, G=1,
Di=1, Hi=112, Wi=112,
Z=1, Y=7, X=7,
stride_d=1, stride_h=2, stride_w=2,
pad_d=0, pad_h=3, pad_w=3,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=512, K=1024, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=128, K=1024, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=128, G=2,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=128, K=256, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=512, K=1024, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=256, K=512, G=1,
Di=1, Hi=14, Wi=14,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=2, pad_w=2,
dilation_d=1, dilation_h=2, dilation_w=2
),
GroupedConvProblem(
N=4, C=64, K=64, G=8,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=512, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=256, K=256, G=256,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=512, K=512, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=512, K=512, G=8,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=512, K=64, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=256, K=128, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=512, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=512, K=64, G=1,
Di=1, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=64, K=64, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=64, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=1024, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=256, K=512, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=128, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=1024, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=64, G=2,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=64, K=128, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=1024, K=512, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=64, G=2,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=128, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=256, K=1024, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=256, K=256, G=256,
Di=1, Hi=16, Wi=16,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=64, G=4,
Di=1, Hi=28, Wi=28,
Z=1, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=256, K=256, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=1024, K=256, G=1,
Di=1, Hi=16, Wi=16,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=32, C=256, K=512, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=128, G=1,
Di=1, Hi=8, Wi=8,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=16, C=64, K=512, G=1,
Di=1, Hi=32, Wi=32,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=256, G=1,
Di=1, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=16, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=128, G=1,
Di=16, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=128, G=1,
Di=32, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=128, G=1,
Di=16, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=128, G=1,
Di=32, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=256, G=1,
Di=8, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=64, K=128, G=1,
Di=16, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=2, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=256, G=1,
Di=16, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=2, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=128, G=1,
Di=32, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=256, G=1,
Di=16, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=256, G=1,
Di=32, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=2, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=256, G=1,
Di=32, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=128, G=1,
Di=16, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=128, G=1,
Di=8, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=256, G=1,
Di=16, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=2, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=256, G=1,
Di=32, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=128, G=1,
Di=16, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=32, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=256, G=1,
Di=32, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=128, G=1,
Di=32, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=8, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=128, G=1,
Di=8, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=256, G=1,
Di=8, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=128, G=1,
Di=32, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=256, G=1,
Di=16, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=64, K=128, G=1,
Di=8, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=256, G=1,
Di=16, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=128, G=1,
Di=16, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=16, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=2, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=128, G=1,
Di=32, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=128, G=1,
Di=32, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=256, G=1,
Di=8, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=128, G=1,
Di=16, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=2, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=128, G=1,
Di=8, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=256, G=1,
Di=32, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=2, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=128, G=1,
Di=32, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=128, G=1,
Di=32, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=128, G=1,
Di=8, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=128, K=128, G=1,
Di=32, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=32, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=2, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=8, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=256, G=1,
Di=16, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=128, G=1,
Di=16, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=256, G=1,
Di=32, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=128, G=1,
Di=8, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=2, C=128, K=256, G=1,
Di=8, Hi=28, Wi=28,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=4, C=128, K=256, G=1,
Di=32, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=1, C=64, K=128, G=1,
Di=32, Hi=28, Wi=28,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=128, K=256, G=1,
Di=32, Hi=56, Wi=56,
Z=1, Y=1, X=1,
stride_d=1, stride_h=1, stride_w=1,
pad_d=0, pad_h=0, pad_w=0,
dilation_d=1, dilation_h=1, dilation_w=1
),
GroupedConvProblem(
N=8, C=64, K=128, G=1,
Di=8, Hi=56, Wi=56,
Z=3, Y=3, X=3,
stride_d=1, stride_h=1, stride_w=1,
pad_d=1, pad_h=1, pad_w=1,
dilation_d=1, dilation_h=1, dilation_w=1
),
]