Hugging Face's logo

Spaces:
sczhou
/
ProPainter
Running on A10G

App Files Community
8
ProPainter
File size: 935 Bytes 
320e465
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
 
{
    "seed": 2023,
    "save_dir": "experiments_model/",
    "train_data_loader": {
        "name": "youtube-vos",
        "video_root": "your_video_root",
        "flow_root": "your_flow_root",
        "w": 432,
        "h": 240,
        "num_local_frames": 10,
        "num_ref_frames": 1,
        "load_flow": 0
    },
    "losses": {
        "flow_weight": 0.25
    },
    "model": {
        "net": "recurrent_flow_completion"
    },
    "trainer": {
        "version": "trainer_flow_w_edge",
        "type": "Adam",
        "beta1": 0,
        "beta2": 0.99,
        "lr": 5e-5,
        "batch_size": 8,
        "num_workers": 4,
        "num_prefetch_queue": 4,
        "log_freq": 100,
        "save_freq": 5e3,
        "iterations": 700e3,
        "scheduler": {
            "type": "MultiStepLR",
            "milestones": [
                300e3, 400e3, 500e3, 600e3
            ],
            "gamma": 0.2
        }
    }
}