上传者: 42103587
|
上传时间: 2022-02-12 19:24:34
|
文件大小: 76KB
|
文件类型: -
CNNIQAplusplus
以下论文的PyTorch 1.3实施:
笔记
在这里,选择优化器作为Adam,而不是本文中带有势头的SGD。
训练
CUDA_VISIBLE_DEVICES=0 python main.py --exp_id=0 --database=LIVE --model=CNNIQAplusplus
训练前, im_dir在config.yaml被指定必须的。
可视化
tensorboard --logdir=tensorboard_logs --port=6006 # in the server (host:port)
ssh -p port -L 6006:localhost:6006 user@host # in your PC. See the visualization in your PC
要求
conda create -n reproducib