08a9c32 3a4e0c0 08a9c32
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import torch lr = 1e-5 path = './weights_CLIP_aesth_7000' save_path = '/home/ryn_mote/Misc/vlm_with_pooled_for_text_genrec/inter-est_CLIP_aesth' epochs = 1 batch_size = 16 device = 'cuda' dtype = torch.bfloat16 max_tokens = 32