文件名称：pytorch_feedback-network:这是feedback-network的pytorch实现（CVPR 2017，Zamir等人）
文件大小：13KB
文件格式：ZIP
更新时间：2024-06-06 21:08:11
pytorch pytorch-implmention feedback-networks Python
纸 这是feedback 的pytorch实现（CVPR 2017，Zamir等人） 要求 火炬= 0.3.1 Python= 2.7 numpy的> = 1.14.2 火车 数据： 下载数据集（二进制文件），并将其放在文件夹./data/ 跑步： 运行classifier_train.py python FeedbackNet_train . py 每10个周期，经过训练的模型将保存在文件夹./models/ 。 注意力： 您可以调整参数batch_size以适合您的GPU内存。 评估 数据： 下载数据集（二进制文件），并将其放在文件夹./data/ 在CIFAR100数据集上评估： 将检查点文件放在文件夹./models/ ，并通过在文件classifier_train.py设置参数params.ckpt来指定检查点文件路径。 您可以下载我的基准模型 运行cla
【文件预览】：
pytorch_feedback-network-master
----FeedbackNet_test.py(733B)
----models()
--------README.md(117B)
----FeedbackNet_train.py(2KB)
----data()
--------__init__.py(1KB)
----.gitignore(1KB)
----README.md(2KB)
----utils()
--------Trainer.py(7KB)
--------Tester.py(3KB)
--------__init__.py(79B)
--------log.py(492B)
----network()
--------convlstmstack.py(1KB)
--------convlstmcell.py(10KB)
--------feedbackmodule.py(2KB)
--------__init__.py(36B)
--------feedbacknet.py(2KB)