train first stage

python unet_train.py --data_dir /data/common/ILSVRC/Data/CLS-LOC/train --log_dir output/cls0.1_tanh --w_cls0 0.1

Name		Name	Last commit message	Last commit date
Latest commit History 99 Commits
CNNDetection @ 195892d		CNNDetection @ 195892d
GANimageDetection @ 0b29b59		GANimageDetection @ 0b29b59
figures		figures
guided-diffusion @ 0b5f5d0		guided-diffusion @ 0b5f5d0
latent-diffusion @ 25234da		latent-diffusion @ 25234da
logger @ 70b189a		logger @ 70b189a
output		output
samples		samples
stylegan2-pytorch @ ea67408		stylegan2-pytorch @ ea67408
.gitignore		.gitignore
.gitmodules		.gitmodules
README.md		README.md
inv_unet_train.py		inv_unet_train.py
psnr_opencv.py		psnr_opencv.py
testall.sh		testall.sh
testall_inv.sh		testall_inv.sh
testall_patch.sh		testall_patch.sh
unet_cls_test.py		unet_cls_test.py
unet_recon_test.py		unet_recon_test.py
unet_train.py		unet_train.py

Provide feedback