English
self-supervised learning
barlow-twins
6 papers
wgcban's picture
Upload 98 files
803ef9e
#!/bin/bash
gpu=0
dataset=cifar100
arch=resnet18
feature_dim=2048
is_mixup=true # true, false
batch_size=256
epochs=2000
lr=0.01
lr_shed=cosine #"step", "cosine" # step, cosine
mixup_loss_scale=4.0 # scale w.r.t. lambda: 0.0078125 * 5 = 0.0390625
lmbda=0.0078125
timestamp=$(date +"%Y%m%d%H%M%S")
session_name="python_session_$timestamp"
echo ${session_name}
screen -dmS "$session_name"
screen -S "$session_name" -X stuff "conda activate ssl-aug^M"
screen -S "$session_name" -X stuff "CUDA_VISIBLE_DEVICES=${gpu} python main.py --lmbda ${lmbda} --corr_zero --batch_size ${batch_size} --feature_dim ${feature_dim} --dataset ${dataset} --is_mixup ${is_mixup} --mixup_loss_scale ${mixup_loss_scale} --epochs ${epochs} --arch ${arch} --gpu ${gpu} --lr_shed ${lr_shed} --lr ${lr}^M"
screen -S "$session_name" -X detach