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: TerraIncognita
	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/5686d05040a042d3a10a3836ba1c25a0
	save_linear_probed_clip: False
	save_model_every_checkpoint: False
	seed: 1135498085
	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.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 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.0562027708  0.0660752039  0.0461376219  0.0421940928  0.0799845937  0.0888546482  0.0557304786  0.0566750630  0.0677714043  0.0671768707  0.0000000000  4.7508945465  2.1692652702  0             1.4753668308 
0.4247481106  0.7271180258  0.8165040865  0.8069620253  0.7581204262  0.7544940935  0.4238035264  0.4256926952  0.6443594646  0.6198979592  3.0226700252  2.6318992523  2.4375615120  300           0.1538426654 
0.5039357680  0.8087713329  0.8919061429  0.8818565401  0.8279625112  0.8310220853  0.4965365239  0.5113350126  0.7582324198  0.7134353741  6.0453400504  2.2540418677  2.4375615120  600           0.1797190038 
0.5196788410  0.8259951862  0.9185341418  0.8934599156  0.8477339838  0.8464304058  0.5179471033  0.5214105793  0.7843637136  0.7380952381  9.0680100756  2.0905665227  2.4375615120  900           0.1829384756 
0.5388853902  0.8459530445  0.9317163195  0.9166666667  0.8627551675  0.8618387262  0.5399874055  0.5377833753  0.8000849798  0.7593537415  12.090680100  1.9809008459  2.4375615120  1200          0.1824604686 
0.5541561710  0.8549598843  0.9454257843  0.9251054852  0.8788034407  0.8685156651  0.5591939547  0.5491183879  0.8270660718  0.7712585034  15.113350125  1.8904190008  2.4375615120  1500          0.1787130022 
0.5724181357  0.8657228321  0.9514895861  0.9324894515  0.8859930671  0.8798151002  0.5780856423  0.5667506297  0.8381134481  0.7848639456  18.136020151  1.7503165797  2.4375615120  1800          0.1822891871 
0.5823362717  0.8695496585  0.9596625363  0.9345991561  0.8947233278  0.8823831536  0.5878463476  0.5768261965  0.8517102188  0.7916666667  21.158690176  1.6882862981  2.4375615120  2100          0.1887124332 
0.5960327453  0.8754934207  0.9630899025  0.9377637131  0.9025548851  0.8885464818  0.5950881612  0.5969773300  0.8591459528  0.8001700680  24.181360201  1.6144486503  2.4375615120  2400          0.1854904850 
0.6068954657  0.8796534439  0.9667809122  0.9388185654  0.9071767878  0.8931689779  0.6004408060  0.6133501259  0.8687061823  0.8069727891  27.204030226  1.5716748035  2.4375615120  2700          0.1833404454 
0.6125629720  0.8914379354  0.9702082784  0.9462025316  0.9105148286  0.9049820236  0.6079974811  0.6171284635  0.8752921181  0.8231292517  30.226700251  1.5503821655  2.4375615120  3000          0.1856874506 
0.6100440803  0.8850912522  0.9717901397  0.9483122363  0.9138528694  0.8957370313  0.6079974811  0.6120906801  0.8784788613  0.8112244898  33.249370277  1.1987510335  5.3968682289  3300          0.2099644574 
0.6105163725  0.8896485187  0.9765357237  0.9440928270  0.9197586340  0.9008731382  0.6102015113  0.6108312343  0.8920756320  0.8239795918  36.272040302  1.0207344393  5.3968682289  3600          0.2199678397 
0.6081549115  0.8980095305  0.9765357237  0.9556962025  0.9233534472  0.9075500770  0.6105163728  0.6057934509  0.8893137880  0.8307823129  39.294710327  0.9957478962  5.3968682289  3900          0.2224543325 
0.6106738032  0.9004867025  0.9804903770  0.9546413502  0.9250224676  0.9049820236  0.6105163728  0.6108312343  0.8975993202  0.8418367347  42.317380352  0.9835699236  5.3968682289  4200          0.2210654926 
0.6113035261  0.9064580957  0.9794358028  0.9609704641  0.9257927847  0.9080636877  0.6117758186  0.6108312343  0.9014234119  0.8503401361  45.340050377  0.9710950436  5.3968682289  4500          0.2201034125 
0.6095717881  0.9018046018  0.9820722383  0.9493670886  0.9293875979  0.9116589625  0.6095717884  0.6095717884  0.9009985129  0.8443877551  48.362720403  0.9661930575  5.3968682289  4800          0.2201824005 
0.6103589418  0.9042582670  0.9812813077  0.9504219409  0.9313133907  0.9137134052  0.6098866499  0.6108312343  0.9075844487  0.8486394558  50.377833753  0.9593510121  5.3968682289  5000          0.2207164073 
