Zhihan Gao

icon indicating a website link https://gaozhihan.github.io/

icon company The Hong Kong University of Science and Technology icon location Hong Kong

Results 22 comments of Zhihan Gao

Question about the 2D CNN+Downsampler

Thank you very much for reporting this issue. You correctly pointed out that the actual model architecture is: ``` Conv3×3 GroupNorm16 LeakyReLU Conv3×3 GroupNorm16 LeakyReLU PatchMerge LayerNorm Linear ``` Our...

Question about the input shape of 'CuboidTransformerEncoder' in cuboid_transformer.py

Thank you for your question. The input shape asserted in [CuboidTransformerEncoder](https://github.com/amazon-science/earth-forecasting-transformer/blob/7732b03bdb366110563516c3502315deab4c2026/src/earthformer/cuboid_transformer/cuboid_transformer.py#L1645) in line 1898 https://github.com/amazon-science/earth-forecasting-transformer/blob/7732b03bdb366110563516c3502315deab4c2026/src/earthformer/cuboid_transformer/cuboid_transformer.py#L1898 should match the shape of `x` in line 3184 https://github.com/amazon-science/earth-forecasting-transformer/blob/7732b03bdb366110563516c3502315deab4c2026/src/earthformer/cuboid_transformer/cuboid_transformer.py#L3184 which has been downsampled by...