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: [2]
Args:
	algorithm: Selective_KD
	checkpoint_freq: 300
	data_dir: ./domainbed/data
	dataset: PACS
	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: 3
	output_dir: sweep/ablation3/outputs/8de41b671cdedc186a7d7176d2ba5a4c
	save_linear_probed_clip: False
	save_model_every_checkpoint: False
	seed: 1418576498
	skip_model_save: False
	steps: 5001
	sweep: True
	task: domain_generalization
	test_envs: [2]
	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.8610090196552951
	lambda2: 0.5777772877463595
	last_k_epoch: 0.32350558703299503
	lr: 5e-05
	nonlinear_classifier: False
	resnet18: False
	resnet_dropout: 0
	student_model: resnet
	temperature: 2.866557071912062
	weight_decay: 0.0001
	worst_case_p: 0.25
using augment transform
using augment transform
using normal 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.1560628742  0.1469599307  0.1403294692  0.1369193154  0.1199360341  0.1217948718  0.1354790419  0.1766467066  0.1895674300  0.1821656051  0.0000000000  8.6642704010  2.0074062347  0             1.4868040085 
0.9898952091  0.9461075434  0.9774252593  0.9486552567  0.9728144989  0.9444444444  0.9887724551  0.9910179641  0.9516539440  0.9452229299  7.1856287425  2.9239098863  2.2566514015  300           0.1505352322 
0.9913922151  0.9563051949  0.9896278218  0.9511002445  0.9920042644  0.9636752137  0.9887724551  0.9940119760  0.9672391858  0.9541401274  14.371257485  1.2322048632  2.2566514015  600           0.1686847663 
0.9913922151  0.9615991855  0.9963392312  0.9584352078  0.9925373134  0.9722222222  0.9887724551  0.9940119760  0.9742366412  0.9541401274  21.556886227  1.0418709091  2.2566514015  900           0.1685720460 
0.9932634726  0.9621471725  0.9963392312  0.9682151589  0.9936034115  0.9615384615  0.9925149701  0.9940119760  0.9802798982  0.9566878981  28.742514970  0.9924350715  2.2566514015  1200          0.1691243140 
0.9925149696  0.9584625604  0.9981696156  0.9584352078  0.9968017058  0.9615384615  0.9910179641  0.9940119760  0.9802798982  0.9554140127  35.928143712  0.9302585991  2.2566514015  1500          0.1737557030 
0.9906437121  0.9665370696  0.9987797437  0.9608801956  0.9978678038  0.9807692308  0.9902694611  0.9910179641  0.9805979644  0.9579617834  43.113772455  0.9005964434  2.2566514015  1800          0.1722271601 
0.9906437121  0.9620238139  0.9975594875  0.9584352078  0.9962686567  0.9722222222  0.9902694611  0.9910179641  0.9834605598  0.9554140127  50.299401197  0.8854282310  2.2566514015  2100          0.1731127524 
0.9902694606  0.9601404608  0.9987797437  0.9511002445  0.9994669510  0.9700854701  0.9895209581  0.9910179641  0.9831424936  0.9592356688  57.485029940  0.8703427915  2.2566514015  2400          0.1735895673 
0.9902694606  0.9642633113  0.9993898719  0.9608801956  0.9989339019  0.9764957265  0.9895209581  0.9910179641  0.9850508906  0.9554140127  64.670658682  0.8426424909  2.2566514015  2700          0.1742927019 
0.9902694606  0.9652631843  0.9993898719  0.9608801956  1.0000000000  0.9807692308  0.9895209581  0.9910179641  0.9856870229  0.9541401274  71.856287425  0.8266216648  2.2566514015  3000          0.1732558521 
0.9906437121  0.9665370319  0.9993898719  0.9706601467  0.9962686567  0.9786324786  0.9902694611  0.9910179641  0.9850508906  0.9503184713  79.041916167  0.8111787069  2.2566514015  3300          0.1733241526 
0.9921407181  0.9612636924  0.9987797437  0.9608801956  0.9984008529  0.9636752137  0.9902694611  0.9940119760  0.9850508906  0.9592356688  86.227544910  0.5898298818  5.3973727226  3600          0.2023564672 
0.9917664666  0.9620923359  0.9987797437  0.9535452323  0.9984008529  0.9722222222  0.9895209581  0.9940119760  0.9850508906  0.9605095541  93.413173652  0.4499277944  5.3973727226  3900          0.2148346710 
0.9928892211  0.9655371966  0.9993898719  0.9608801956  0.9989339019  0.9764957265  0.9917664671  0.9940119760  0.9904580153  0.9592356688  100.59880239  0.4212967677  5.3973727226  4200          0.2101523137 
0.9928892211  0.9649276911  0.9993898719  0.9633251834  0.9989339019  0.9722222222  0.9917664671  0.9940119760  0.9860050891  0.9592356688  107.78443113  0.4031605853  5.3973727226  4500          0.2155978664 
0.9932634726  0.9699340595  1.0000000000  0.9706601467  0.9989339019  0.9786324786  0.9925149701  0.9940119760  0.9879134860  0.9605095541  114.97005988  0.3984129972  5.3973727226  4800          0.2182458536 
0.9932634726  0.9659960484  0.9993898719  0.9682151589  0.9994669510  0.9743589744  0.9925149701  0.9940119760  0.9866412214  0.9554140127  119.76047904  0.3846927273  5.3973727226  5000          0.2207409859 
