无规范化网络和SGD的Tensorflow实现-自适应梯度剪切
论文: :
原始代码: :
安装及使用
我建议使用Docker运行代码:
docker build -t nfnets/imagenet:latest --build-arg USER_ID=$(id -u) --build-arg GROUP_ID=$(id -g) .
要在imagenet数据集上训练NFNet,请执行以下操作:
docker run --rm -it --gpus all -v $(pwd):/tf -p 8889:8888 -p 6006:6006 nfnets/imagenet:latest python train.py --variant F0 --batch_size 4096 --num_epochs 360
请参阅train.py模块以获取更多参数。
预先训练的权重已转换为
1