Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,14 @@ The first pretext task is **CURTAIN** (occupancy / light-front forecast). The
architecture is built so a new self-supervised method is a small plugin under
`pretext/`, reusing the data, backbone, and training engine unchanged.

```mermaid
flowchart LR
D["your data<br/>reader + geometry"] --> T["PretextTask<br/>e.g. CURTAIN"]
T --> B["Backbone<br/>e.g. DeepIce"]
B --> H["objective heads<br/>+ loss"]
H --> X["pretrained backbone<br/>for your fine tune"]
```

## 馃Л Philosophy
The core is framework-agnostic and fits neatly into plain PyTorch: it depends
only on torch, pytorch-lightning and numpy. Readers are ordinary indexable
Expand Down
Loading