File size: 233 Bytes
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