|
#!/bin/bash |
|
|
|
gpu=0 |
|
dataset=cifar10 |
|
arch=resnet18 |
|
feature_dim=1024 |
|
is_mixup=true |
|
batch_size=256 |
|
epochs=2000 |
|
lr=0.01 |
|
lr_shed=cosine |
|
mixup_loss_scale=4.0 |
|
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 |
