|
|
|
|
lint:
|
|
|
|
|
black .
|
|
|
|
|
isort --profile=black *.py
|
|
|
|
|
flake8 --ignore E501,W503 *.py
|
|
|
|
|
|
|
|
|
|
test:
|
|
|
|
|
# python main.py --alpha 1 --lr 1e-2 --max_epochs 200 --bs 256 --seed 856 --width 2048
|
|
|
|
|
python newmain.py fit \
|
|
|
|
|
--seed_everything 21 \
|
|
|
|
|
--data.batch_size 256 \
|
|
|
|
|
--data.train_size 0 \
|
|
|
|
|
--data.val_size 100000 \
|
|
|
|
|
--model.alpha 0 \
|
|
|
|
|
--model.width 2048 \
|
|
|
|
|
--trainer.fast_dev_run 1 \
|
|
|
|
|
--trainer.min_epochs 1 \
|
|
|
|
|
--trainer.max_epochs 10 \
|
|
|
|
|
--trainer.check_val_every_n_epoch 1 \
|
|
|
|
|
--trainer.callbacks callbacks.SaveImageCallback \
|
|
|
|
|
--trainer.callbacks.init_args.final_dir out \
|
|
|
|
|
--trainer.callbacks.init_args.save_interval 0 \
|
|
|
|
|
--optimizer torch.optim.Adam \
|
|
|
|
|
--optimizer.init_args.lr 0.01 \
|
|
|
|
|
--lr_scheduler lightning.pytorch.cli.ReduceLROnPlateau \
|
|
|
|
|
--lr_scheduler.init_args.patience 5 \
|
|
|
|
|
--lr_scheduler.init_args.cooldown 10 \
|
|
|
|
|
--lr_scheduler.init_args.factor 0.05 \
|
|
|
|
|
--lr_scheduler.init_args.verbose true \
|
|
|
|
|
--print_config
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
help:
|
|
|
|
|
# python newmain.py fit --help --trainer.callbacks.help
|
|
|
|
|
# python newmain.py fit --lr_scheduler.help lightning.pytorch.cli.ReduceLROnPlateau
|
|
|
|
|
python newmain.py fit --help
|
|
|
|
|
|
|
|
|
|
search:
|
|
|
|
|
python newsearch.py
|
|
|
|
|
|
|
|
|
|
hsv:
|
|
|
|
|
python hsv.py
|
|
|
|
|
|
|
|
|
|
animate:
|
|
|
|
|
ffmpeg -i lightning_logs/version_258/e%04d.png \
|
|
|
|
|
-c:v libx264 \
|
|
|
|
|
-vf "fps=12,format=yuv420p,pad=ceil(iw/2)*2:ceil(ih/2)*2" \
|
|
|
|
|
~/animated.mp4
|
|
|
|
|
|
|
|
|
|
clean:
|
|
|
|
|
rm -rf lightning_logs
|
|
|
|
|
rm -f out/*.png
|
|
|
|
|
rm -rf __pycache__/
|
|
|
|
|
cp hsv.png out/
|
