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: 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: 4
	output_dir: sweep/ablation3/outputs/ce6af83c63f0d6e42b9d41566ddc642e
	save_linear_probed_clip: False
	save_model_every_checkpoint: False
	seed: 916426713
	skip_model_save: False
	steps: 5001
	sweep: True
	task: domain_generalization
	test_envs: [1]
	trial_seed: 0
	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.8001163740830898
	lambda2: 0.7340462818445315
	last_k_epoch: 0.3149495809125332
	lr: 3e-05
	nonlinear_classifier: False
	resnet18: False
	resnet_dropout: 0
	student_model: resnet
	temperature: 2.8284464949429635
	weight_decay: 1e-06
	worst_case_p: 0.2
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.1785030889  0.1288640136  0.1446003661  0.1222493888  0.1711087420  0.1858974359  0.1377245509  0.0898203593  0.1946564885  0.1745222930  0.0000000000  9.0298347473  1.7945718765  0             1.5014474392 
0.7841493080  0.9553897574  0.9841366687  0.9486552567  0.7862473348  0.7820512821  0.9992514970  0.9850299401  0.9472010178  0.9324840764  7.1856287425  3.2078731541  2.0946040154  300           0.1461566083 
0.8151048781  0.9681961879  0.9914582062  0.9559902200  0.8118336887  0.8183760684  1.0000000000  0.9970059880  0.9605597964  0.9515923567  14.371257485  1.3158038529  2.0946040154  600           0.1629046631 
0.8268501809  0.9670494368  0.9957291031  0.9559902200  0.8182302772  0.8354700855  1.0000000000  0.9910179641  0.9672391858  0.9541401274  21.556886227  1.1499490307  2.0946040154  900           0.1635959148 
0.8313833757  0.9718808001  0.9945088469  0.9657701711  0.8251599147  0.8376068376  1.0000000000  0.9970059880  0.9732824427  0.9528662420  28.742514970  1.0629441508  2.0946040154  1200          0.1637526774 
0.8388528963  0.9687479506  0.9951189750  0.9559902200  0.8336886994  0.8440170940  1.0000000000  0.9910179641  0.9748727735  0.9592356688  35.928143712  1.0197895656  2.0946040154  1500          0.1640312727 
0.8433883691  0.9645143143  0.9975594875  0.9535452323  0.8384861407  0.8482905983  1.0000000000  0.9820359281  0.9761450382  0.9579617834  43.113772455  0.9999095118  2.0946040154  1800          0.1639766328 
0.8511289701  0.9696774327  0.9981696156  0.9608801956  0.8432835821  0.8589743590  1.0000000000  0.9940119760  0.9761450382  0.9541401274  50.299401197  0.9409087592  2.0946040154  2100          0.1642271407 
0.8527326732  0.9679915670  0.9969493594  0.9584352078  0.8422174840  0.8632478632  1.0000000000  0.9850299401  0.9790076336  0.9605095541  57.485029940  0.9323854403  2.0946040154  2400          0.1647047710 
0.8546006228  0.9717663140  0.9975594875  0.9633251834  0.8438166311  0.8653846154  1.0000000000  0.9940119760  0.9780534351  0.9579617834  64.670658682  0.9026699905  2.0946040154  2700          0.1643757804 
0.8564685725  0.9742798237  0.9981696156  0.9657701711  0.8454157783  0.8675213675  0.9992514970  0.9940119760  0.9834605598  0.9630573248  71.856287425  0.8903063854  2.0946040154  3000          0.1646433377 
0.8529969197  0.9752093057  0.9987797437  0.9706601467  0.8448827292  0.8611111111  1.0000000000  0.9970059880  0.9812340967  0.9579617834  79.041916167  0.8790466428  2.0946040154  3300          0.1641156109 
0.8551336719  0.9709513180  1.0000000000  0.9608801956  0.8448827292  0.8653846154  1.0000000000  0.9940119760  0.9844147583  0.9579617834  86.227544910  0.6675626122  5.4003272057  3600          0.1894447271 
0.8567350970  0.9713416855  0.9981696156  0.9633251834  0.8459488273  0.8675213675  1.0000000000  0.9940119760  0.9853689567  0.9566878981  93.413173652  0.4463499792  5.4003272057  3900          0.2060889117 
0.8596691447  0.9707683100  0.9993898719  0.9633251834  0.8496801706  0.8696581197  1.0000000000  0.9910179641  0.9837786260  0.9579617834  100.59880239  0.4112648748  5.4003272057  4200          0.2059652948 
0.8620724213  0.9706195629  0.9993898719  0.9633251834  0.8502132196  0.8739316239  1.0000000000  0.9880239521  0.9860050891  0.9605095541  107.78443113  0.3916730424  5.4003272057  4500          0.2058061179 
0.8618058968  0.9701264125  1.0000000000  0.9682151589  0.8496801706  0.8739316239  1.0000000000  0.9880239521  0.9844147583  0.9541401274  114.97005988  0.3822848330  5.4003272057  4800          0.2062640452 
0.8602044717  0.9681403141  0.9963392312  0.9584352078  0.8486140725  0.8717948718  1.0000000000  0.9880239521  0.9847328244  0.9579617834  119.76047904  0.3601064710  5.4003272057  5000          0.2100175881 
