diff options
| author | tingchunw <tingchunw@nvidia.com> | 2017-12-04 16:52:46 -0800 |
|---|---|---|
| committer | tingchunw <tingchunw@nvidia.com> | 2017-12-04 16:52:46 -0800 |
| commit | 9054cf9b0c327a5077fd0793abe178f400da3315 (patch) | |
| tree | 3c69c07bdcba86c47d8442648fd69c0434e04136 /scripts | |
| parent | f9e9999541d67a908a169cc88407675133130e1f (diff) | |
first commit
Diffstat (limited to 'scripts')
| -rwxr-xr-x | scripts/test_1024p.sh | 3 | ||||
| -rwxr-xr-x | scripts/test_1024p_feat.sh | 5 | ||||
| -rwxr-xr-x | scripts/test_512p.sh | 3 | ||||
| -rwxr-xr-x | scripts/test_512p_feat.sh | 5 | ||||
| -rwxr-xr-x | scripts/train_1024p_12G.sh | 4 | ||||
| -rwxr-xr-x | scripts/train_1024p_24G.sh | 4 | ||||
| -rwxr-xr-x | scripts/train_1024p_feat_12G.sh | 6 | ||||
| -rwxr-xr-x | scripts/train_1024p_feat_24G.sh | 6 | ||||
| -rwxr-xr-x | scripts/train_512p.sh | 2 | ||||
| -rwxr-xr-x | scripts/train_512p_feat.sh | 2 | ||||
| -rwxr-xr-x | scripts/train_512p_multigpu.sh | 2 |
11 files changed, 42 insertions, 0 deletions
diff --git a/scripts/test_1024p.sh b/scripts/test_1024p.sh new file mode 100755 index 0000000..99c1e24 --- /dev/null +++ b/scripts/test_1024p.sh @@ -0,0 +1,3 @@ +################################ Testing ################################
+# labels only
+python test.py --name label2city_1024p --netG local --ngf 32 --resize_or_crop none
\ No newline at end of file diff --git a/scripts/test_1024p_feat.sh b/scripts/test_1024p_feat.sh new file mode 100755 index 0000000..2f4ba17 --- /dev/null +++ b/scripts/test_1024p_feat.sh @@ -0,0 +1,5 @@ +################################ Testing ################################
+# first precompute and cluster all features
+python encode_features.py --name label2city_1024p_feat --netG local --ngf 32 --resize_or_crop none;
+# use instance-wise features
+python test.py --name label2city_1024p_feat ---netG local --ngf 32 --resize_or_crop none --instance_feat
\ No newline at end of file diff --git a/scripts/test_512p.sh b/scripts/test_512p.sh new file mode 100755 index 0000000..3131043 --- /dev/null +++ b/scripts/test_512p.sh @@ -0,0 +1,3 @@ +################################ Testing ################################
+# labels only
+python test.py --name label2city_512p
\ No newline at end of file diff --git a/scripts/test_512p_feat.sh b/scripts/test_512p_feat.sh new file mode 100755 index 0000000..8f25e9c --- /dev/null +++ b/scripts/test_512p_feat.sh @@ -0,0 +1,5 @@ +################################ Testing ################################
+# first precompute and cluster all features
+python encode_features.py --name label2city_512p_feat;
+# use instance-wise features
+python test.py --name label2city_512p_feat --instance_feat
\ No newline at end of file diff --git a/scripts/train_1024p_12G.sh b/scripts/train_1024p_12G.sh new file mode 100755 index 0000000..d5ea7d7 --- /dev/null +++ b/scripts/train_1024p_12G.sh @@ -0,0 +1,4 @@ +############## To train images at 2048 x 1024 resolution after training 1024 x 512 resolution models #############
+##### Using GPUs with 12G memory (not tested)
+# Using labels only
+python train.py --name label2city_1024p --netG local --ngf 32 --num_D 3 --load_pretrain checkpoints/label2city_512p/ --niter_fix_global 20 --resize_or_crop crop --fineSize 1024
\ No newline at end of file diff --git a/scripts/train_1024p_24G.sh b/scripts/train_1024p_24G.sh new file mode 100755 index 0000000..88e58f7 --- /dev/null +++ b/scripts/train_1024p_24G.sh @@ -0,0 +1,4 @@ +############## To train images at 2048 x 1024 resolution after training 1024 x 512 resolution models #############
+######## Using GPUs with 24G memory
+# Using labels only
+python train.py --name label2city_1024p --netG local --ngf 32 --num_D 3 --load_pretrain checkpoints/label2city_512p/ --niter 50 --niter_decay 50 --niter_fix_global 10 --resize_or_crop none
\ No newline at end of file diff --git a/scripts/train_1024p_feat_12G.sh b/scripts/train_1024p_feat_12G.sh new file mode 100755 index 0000000..f8e3d61 --- /dev/null +++ b/scripts/train_1024p_feat_12G.sh @@ -0,0 +1,6 @@ +############## To train images at 2048 x 1024 resolution after training 1024 x 512 resolution models #############
+##### Using GPUs with 12G memory (not tested)
+# First precompute feature maps and save them
+python precompute_feature_maps.py --name label2city_512p_feat;
+# Adding instances and encoded features
+python train.py --name label2city_1024p_feat --netG local --ngf 32 --num_D 3 --load_pretrain checkpoints/label2city_512p_feat/ --niter_fix_global 20 --resize_or_crop crop --fineSize 896 --instance_feat --load_features
\ No newline at end of file diff --git a/scripts/train_1024p_feat_24G.sh b/scripts/train_1024p_feat_24G.sh new file mode 100755 index 0000000..399d720 --- /dev/null +++ b/scripts/train_1024p_feat_24G.sh @@ -0,0 +1,6 @@ +############## To train images at 2048 x 1024 resolution after training 1024 x 512 resolution models #############
+######## Using GPUs with 24G memory
+# First precompute feature maps and save them
+python precompute_feature_maps.py --name label2city_512p_feat;
+# Adding instances and encoded features
+python train.py --name label2city_1024p_feat --netG local --ngf 32 --num_D 3 --load_pretrain checkpoints/label2city_512p_feat/ --niter 50 --niter_decay 50 --niter_fix_global 10 --resize_or_crop none --instance_feat --load_features
\ No newline at end of file diff --git a/scripts/train_512p.sh b/scripts/train_512p.sh new file mode 100755 index 0000000..222c348 --- /dev/null +++ b/scripts/train_512p.sh @@ -0,0 +1,2 @@ +### Using labels only
+python train.py --name label2city_512p
\ No newline at end of file diff --git a/scripts/train_512p_feat.sh b/scripts/train_512p_feat.sh new file mode 100755 index 0000000..9d4859c --- /dev/null +++ b/scripts/train_512p_feat.sh @@ -0,0 +1,2 @@ +### Adding instances and encoded features
+python train.py --name label2city_512p_feat --instance_feat
\ No newline at end of file diff --git a/scripts/train_512p_multigpu.sh b/scripts/train_512p_multigpu.sh new file mode 100755 index 0000000..16f0a1a --- /dev/null +++ b/scripts/train_512p_multigpu.sh @@ -0,0 +1,2 @@ +######## Multi-GPU training example #######
+python train.py --name label2city_512p --batchSize 8 --gpu_ids 0,1,2,3,4,5,6,7
\ No newline at end of file |
