Rename P3D -> TCF (3D turbulent channel flow)

README.md

@@ -14,14 +14,14 @@ for Neural PDE Solvers**. See the code at https://github.com/Howw-Way/ReViT.
 
 | Model | Channels | Grid | Patch | Architecture |
 |-------|----------|------|-------|--------------|
-| `P3D` | 4 | 96³ | 3 | `ReViT3D` |
+| `TCF` | 4 | 96³ | 3 | `ReViT3D` |
 | `MHD_64` | 7 | 64³ | 2 | `ReViT3D` |
 
 ## Usage
 
 ```python
 from revit import load_pretrained
-model = load_pretrained("P3D")        # or "MHD_64"
+model = load_pretrained("TCF")        # or "MHD_64"
 ```
 
 Each subfolder holds `config.json` (the model config), `best_model.pth`

TCF/best_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:20b60310b1842fd2fe6695cddd68a8e526b50c782d378ff1a8e49828a8ccaf91
+size 9513971

TCF/config.json

@@ -0,0 +1,37 @@
+{
+  "additional_vector_chans": 0,
+  "aggregation_method": "custom_attention",
+  "attn_drop_rate": 0.1,
+  "blur_sigma": null,
+  "decoder_type": "trilinear",
+  "depth": [
+    1,
+    2,
+    4,
+    2,
+    1
+  ],
+  "drop_path_rate": 0.1,
+  "drop_rate": 0.1,
+  "embed_dim": 48,
+  "embed_type": "AOVC",
+  "euler_connection": true,
+  "img_D": 96,
+  "img_H": 96,
+  "img_W": 96,
+  "in_channels": 4,
+  "intra_patch_pos_encoding": false,
+  "local_attention": false,
+  "max_embed_dim": 192,
+  "mlp_ratio": 4.0,
+  "num_heads": 8,
+  "out_channels": 4,
+  "patch_merge": "global",
+  "patch_size": 3,
+  "periodic": true,
+  "pos_encoding_type": "eq_rel_simple",
+  "primary_vector_chans": 3,
+  "primary_vector_start_idx": 0,
+  "scalar_chans": 1,
+  "window_size": 2
+}

TCF/config.yaml

@@ -0,0 +1,131 @@
+accumulate_grad_batches: 1
+additional_vector_chans: 0
+aggregation_method: custom_attention
+alpha: 0.99
+attn_drop_rate: 0.1
+augment_angles: null
+augment_name: NoAug
+batch_size: 1
+betas: !!python/tuple
+- 0.9
+- 0.999
+blur_sigma: null
+case_name: mstep1_test1_lr0.001_wd0.001_batch1_seed777_drop0.1_embeddim48_depth5_head8_patch3_AdamW_ReduceLROnPlateau_eulerTrue_sigmaNone_NoAug_PeriTrue_pmglobal
+channels: all
+check_train: false
+decoder_type: trilinear
+depth: &id001
+- 1
+- 2
+- 4
+- 2
+- 1
+device: cuda:3
+drop_path_rate: 0.1
+drop_rate: 0.1
+embed_dim: 48
+embed_type: AOVC
+epochs: 1000
+eps: 1.0e-08
+euler_connection: true
+exclude_filters: null
+img_D: 96
+img_H: 96
+img_W: 96
+img_size: 96
+in_channels: 4
+include_filters: null
+intra_patch_pos_encoding: false
+load_ckpt: false
+local_attention: false
+lr: 0.001
+max_embed_dim: 192
+mlp_ratio: 4.0
+model_config:
+  additional_vector_chans: 0
+  aggregation_method: custom_attention
+  attn_drop_rate: 0.1
+  blur_sigma: null
+  decoder_type: trilinear
+  depth: *id001
+  drop_path_rate: 0.1
+  drop_rate: 0.1
+  embed_dim: 48
+  embed_type: AOVC
+  euler_connection: true
+  img_D: 96
+  img_H: 96
+  img_W: 96
+  in_channels: 4
+  intra_patch_pos_encoding: false
+  local_attention: false
+  max_embed_dim: 192
+  mlp_ratio: 4.0
+  num_heads: 8
+  out_channels: 4
+  patch_merge: global
+  patch_size: 3
+  periodic: true
+  pos_encoding_type: eq_rel_simple
+  primary_vector_chans: 3
+  primary_vector_start_idx: 0
+  scalar_chans: 1
+  window_size: 2
+model_id: null
+model_name: 3D|Emb|AOVC|PE|eq_rel_simple|WS2|LoAFalse|PS3|H48|MH192|D5|NH8|Aggcustom_attention|Dec|trilinear
+model_type: ReViT3D
+momentum: 0.9
+n_steps_input: 1
+n_steps_output: 20
+num_heads: 8
+optimizer: AdamW
+out_channels: 4
+patch_merge: global
+patch_size: 3
+periodic: true
+pos_encoding_type: eq_rel_simple
+primary_vector_chans: 3
+primary_vector_start_idx: 0
+scalar_chans: 1
+scheduler: ReduceLROnPlateau
+seed: 777
+sim_indices:
+- 0
+spatial_slice: null
+task: P3D
+task_dir: ./Results/P3D/ReViT3D/3D|Emb|AOVC|PE|eq_rel_simple|WS2|LoAFalse|PS3|H48|MH192|D5|NH8|Aggcustom_attention|Dec|trilinear/
+test_angles:
+- x_90
+- y_90
+- z_90
+- identity
+- x_180
+- x_270
+- y_180
+- y_270
+- z_180
+- z_270
+- x_90_z_90
+- z_270_x_270
+- x_90_z_270
+- z_90_x_270
+- x_270_z_90
+- z_270_x_90
+- x_270_z_270
+- z_90_x_90
+- x_90_y_90
+- x_90_y_270
+- x_270_y_90
+- x_270_y_270
+- y_90_z_180
+- y_270_z_180
+test_data_path: /mnt/data3/Hao/Solver-Teacher/Dataset/P3D/cropped
+test_steps: 1
+train_data_path: /mnt/data3/Hao/Solver-Teacher/Dataset/P3D/cropped
+tune_metrics: correlation
+unroll_steps: 1
+use_torch_compile: true
+val_split_step: 20
+weight_decay: 0.001
+window_size: 2
+writer_log_dir: logs/train/Results/P3D/ReViT3D/3D|Emb|AOVC|PE|eq_rel_simple|WS2|LoAFalse|PS3|H48|MH192|D5|NH8|Aggcustom_attention|Dec|trilinear/260628-045642-780-07_mstep1_test1_lr0.001_wd0.001_batch1_seed777_drop0.1_embeddim48_depth5_head8_patch3_AdamW_ReduceLROnPlateau_eulerTrue_sigmaNone_NoAug_PeriTrue_pmglobal