CT-transformer/makefile at main · CosmoBean/CT-transformer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
.PHONY: install clean test test-models train-efficientnet train-resnet train-vit train-swin train-flare train-flare-hybrid train-flare-multiscale train-flare-attn-pool train-autoencoder train-vae train-all

install:
	bash scripts/install.sh

clean:
	rm -rf .venv uv.lock main.py pyproject.toml
	rm -rf cache

# Testing commands
test:
	python scripts/test_setup.py

test-models:
	python scripts/validate_setup.py

# Training commands for different models
train-efficientnet:
	python scripts/train.py --model efficientnet_b3 --epochs 10

train-resnet:
	python scripts/train.py --model resnet50 --epochs 10

train-vit:
	python scripts/train.py --model vit_base --epochs 10

train-swin:
	python scripts/train.py --model swin_base_patch4_window7_224 --epochs 10

train-flare:
	python scripts/train.py --model flare --epochs 10

train-flare-hybrid:
	python scripts/train.py --model flare_hybrid --epochs 10

train-flare-multiscale:
	python scripts/train.py --model flare_multiscale --epochs 10

train-flare-attn-pool:
	python scripts/train.py --model flare_attn_pool --epochs 10

train-autoencoder:
	python scripts/train.py --model autoencoder --epochs 20

train-vae:
	python scripts/train.py --model vae --epochs 20

# Quick test runs (1 epoch)
test-efficientnet:
	python scripts/train.py --model efficientnet_b3 --epochs 1

test-resnet:
	python scripts/train.py --model resnet50 --epochs 1

test-vit:
	python scripts/train.py --model vit_base --epochs 1

test-swin:
	python scripts/train.py --model swin_base_patch4_window7_224 --epochs 1

test-flare:
	python scripts/train.py --model flare --epochs 1

test-flare-hybrid:
	python scripts/train.py --model flare_hybrid --epochs 1

test-flare-multiscale:
	python scripts/train.py --model flare_multiscale --epochs 1

test-flare-attn-pool:
	python scripts/train.py --model flare_attn_pool --epochs 1

# Train all models and log results
train-all:
	python scripts/train_all_models.py

# Train all models in background (for overnight runs)
train-all-bg:
	nohup python scripts/train_all_models.py > experiments/training_nohup.log 2>&1 &
	echo "Training started in background. PID: $$!"
	echo "Monitor with: tail -f experiments/training_nohup.log"
	echo "View results with: make view-results"

# View training results
view-results:
	python scripts/view_results.py