diff --git a/egs/audioset/run.sh b/egs/audioset/run.sh index 99dd95b..524c434 100644 --- a/egs/audioset/run.sh +++ b/egs/audioset/run.sh @@ -19,7 +19,7 @@ export TORCH_HOME=../../pretrained_models model=ast dataset=audioset # full or balanced for audioset -set=balanced +set=full imagenetpretrain=True if [ $set == balanced ] then