Environment:
	Python: 3.10.11
	PyTorch: 2.0.1
	Torchvision: 0.15.2
	CUDA: 11.7
	CUDNN: 8500
	NumPy: 1.24.3
	PIL: 9.4.0
	Testing environment: [1]
Args:
	algorithm: Selective_KD
	checkpoint_freq: 300
	data_dir: ./domainbed/data
	dataset: VLCS
	holdout_fraction: 0.2
	hparams: {
    "resnet18": false,
    "resnet_dropout": 0,
    "nonlinear_classifier": false,
    "data_augmentation": true,
    "clip_backbone": "ViT-B/32",
    "student_model": "resnet",
    "SMA": true,
    "batch_size": 32
}
	hparams_seed: 2
	output_dir: sweep/ablation3/outputs/5526ca390179104d26d0950a2cb81dbe
	save_linear_probed_clip: False
	save_model_every_checkpoint: False
	seed: 929210059
	skip_model_save: False
	steps: 5001
	sweep: True
	task: domain_generalization
	test_envs: [1]
	trial_seed: 1
	uda_holdout_fraction: 0
	visualize: False
Not saving models
HParams:
	SMA: True
	batch_size: 32
	class_balanced: False
	clip_backbone: ViT-B/32
	data_augmentation: True
	lambda1: 0.5439617198173775
	lambda2: 0.5509403872292429
	last_k_epoch: 0.38252238504986713
	lr: 5e-05
	nonlinear_classifier: False
	resnet18: False
	resnet_dropout: 0
	student_model: resnet
	temperature: 4.920147743151585
	weight_decay: 1e-06
	worst_case_p: 0.3
using augment transform
using normal transform
using augment transform
using augment transform
using device:  cuda
Using ViT-B/32...
constructing student model
using resnet 50
Using SMA
n_steps 5001
checkpoint_freq 300
agg_test_acc  agg_val_acc   env0_in_acc   env0_out_acc  env1_in_acc   env1_out_acc  env2_in_acc   env2_out_acc  env3_in_acc   env3_out_acc  epoch         loss          mem_gb        step          step_time    
0.2951270631  0.1877925009  0.1669611307  0.1625441696  0.3002352941  0.2900188324  0.1953541508  0.1875000000  0.1828952240  0.2133333333  0.0000000000  4.8578467369  2.1691374779  0             1.5192790031 
0.6596840586  0.9091112130  1.0000000000  0.9964664311  0.6621176471  0.6572504708  0.8971820259  0.8597560976  0.9374305813  0.8711111111  8.4805653710  1.4304412446  2.4361162186  300           0.1581060799 
0.6573315605  0.9128859312  1.0000000000  0.9929328622  0.6555294118  0.6591337100  0.9341203351  0.8612804878  0.9622362088  0.8844444444  16.961130742  0.8888524280  2.4361162186  600           0.1854963144 
0.6545075880  0.9100559585  1.0000000000  0.9964664311  0.6517647059  0.6572504708  0.9611576542  0.8551829268  0.9759348389  0.8785185185  25.441696113  0.7590882639  2.4361162186  900           0.1832239795 
0.6500361136  0.9125680031  1.0000000000  0.9964664311  0.6465882353  0.6534839925  0.9729626809  0.8597560976  0.9825990374  0.8814814815  33.922261484  0.6769049044  2.4361162186  1200          0.1870384924 
0.6448578705  0.9095907372  1.0000000000  0.9964664311  0.6437647059  0.6459510358  0.9832444783  0.8582317073  0.9881525361  0.8740740741  42.402826855  0.6445118544  2.4361162186  1500          0.1850423630 
0.6436813999  0.9142253836  1.0000000000  1.0000000000  0.6414117647  0.6459510358  0.9859101295  0.8582317073  0.9896334691  0.8844444444  50.883392226  0.6210656539  2.4361162186  1800          0.1816561230 
0.6443886116  0.9181760009  1.0000000000  1.0000000000  0.6371764706  0.6516007533  0.9866717441  0.8582317073  0.9918548686  0.8962962963  59.363957597  0.6143318245  2.4361162186  2100          0.1826407552 
0.6453293450  0.9100416556  1.0000000000  0.9964664311  0.6409411765  0.6497175141  0.9935262757  0.8536585366  0.9940762680  0.8800000000  67.844522968  0.6308748550  2.4361162186  2400          0.1831315223 
0.6455646391  0.9160677020  1.0000000000  0.9964664311  0.6414117647  0.6497175141  0.9927646611  0.8643292683  0.9970381340  0.8874074074  76.325088339  0.5907191387  2.4361162186  2700          0.1822983487 
0.6448583136  0.9157211680  1.0000000000  1.0000000000  0.6418823529  0.6478342750  0.9931454684  0.8597560976  0.9970381340  0.8874074074  84.805653710  0.6029710038  2.4361162186  3000          0.1810918291 
0.6441519882  0.9107113818  1.0000000000  1.0000000000  0.6423529412  0.6459510358  0.9950495050  0.8521341463  0.9962976675  0.8800000000  93.286219081  0.4337191028  5.3990564346  3300          0.2097485534 
0.6441515450  0.9122643779  1.0000000000  1.0000000000  0.6442352941  0.6440677966  0.9965727342  0.8597560976  0.9966679008  0.8770370370  101.76678445  0.3303081816  5.3990564346  3600          0.2185617646 
0.6443872823  0.9162149952  1.0000000000  1.0000000000  0.6428235294  0.6459510358  0.9954303123  0.8597560976  0.9966679008  0.8888888889  110.24734982  0.3138469093  5.3990564346  3900          0.2224520858 
0.6425044862  0.9127295992  1.0000000000  1.0000000000  0.6409411765  0.6440677966  0.9977151561  0.8567073171  0.9970381340  0.8814814815  118.72791519  0.3064382042  5.3990564346  4200          0.2159732533 
0.6415628666  0.9130618302  1.0000000000  0.9964664311  0.6409411765  0.6421845574  0.9961919269  0.8597560976  0.9988893003  0.8829629630  127.20848056  0.3005206291  5.3990564346  4500          0.2161182173 
0.6417981608  0.9200898338  1.0000000000  0.9964664311  0.6414117647  0.6421845574  0.9977151561  0.8719512195  0.9988893003  0.8918518519  135.68904593  0.2951520611  5.3990564346  4800          0.2147378747 
0.6410922784  0.9139922729  1.0000000000  0.9964664311  0.6400000000  0.6421845574  0.9980959634  0.8536585366  0.9981488338  0.8918518519  141.34275618  0.2931837013  5.3990564346  5000          0.2162805545 
